diff --git a/klee_stm_gdb.py b/klee_stm_gdb.py
index 3b38aebb9f20eeedf5b9aebc56d43f8a0992a68a..296d3e81fe612f9d1b3c0e33bc8cfc6c82c1661b 100644
--- a/klee_stm_gdb.py
+++ b/klee_stm_gdb.py
@@ -275,9 +275,15 @@ def ktest_setdata(file_index):
 
 
 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 the "klee-last" folder """
     for filename in os.listdir(directory):
         if filename.endswith(".ktest"):
             file_list.append(os.path.join(directory, filename))