diff --git a/tests/InferenceTest.cpp b/tests/InferenceTest.cpp
index 89e78de..cf97459 100644
--- a/tests/InferenceTest.cpp
+++ b/tests/InferenceTest.cpp
@@ -92,6 +92,12 @@
 
 bool ValidateDirectory(std::string& dir)
 {
+    if (dir.empty())
+    {
+        std::cerr << "No directory specified" << std::endl;
+        return false;
+    }
+
     if (dir[dir.length() - 1] != '/')
     {
         dir += "/";
@@ -103,6 +109,12 @@
         return false;
     }
 
+    if (!boost::filesystem::is_directory(dir))
+    {
+        std::cerr << "Given directory [" << dir << "] is not a directory" << std::endl;
+        return false;
+    }
+
     return true;
 }