diff --git a/profiling_db_parser.py b/profiling_db_parser.py
index 23d32a658a7e19d9d4202218a6f921e4e32d75fa..c06c6be387caddb34926c0a51e1de9933ee86622 100644
--- a/profiling_db_parser.py
+++ b/profiling_db_parser.py
@@ -120,9 +120,15 @@ def ktest_setdata(filename):
 
 
 def ktest_iterate():
-    """ Iterate over all files ending with ktest in klee-last folder """
+    """ Get the list of folders in current directory, sort and then grab the
+        last one.
+    """
+    dirlist = next(os.walk('.'))[1]
+    dirlist.sort()
+    directory = dirlist[-1]
+    print("Directory: " + directory)
 
-    directory = "klee-last"
+    """ Iterate over all files ending with ktest in "klee-last" folder """
     for filename in os.listdir(directory):
         if filename.endswith(".ktest"):
             file_list.append(os.path.join(directory, filename))