diff --git a/api/GenerateTestFiles.cpp b/api/GenerateTestFiles.cpp
index f70c444..529bd2a 100644
--- a/api/GenerateTestFiles.cpp
+++ b/api/GenerateTestFiles.cpp
@@ -1030,11 +1030,9 @@
 
 bool GenerateTestFiles(const string& directory, int versionOfTestFiles) {
     bool success = true;
-    for (auto specFile : systemSpecification.getSpecFiles()) {
-        for (auto f : specFile->getFunctionsMap()) {
-            if (!writeTestFilesForFunction(*f.second, directory, versionOfTestFiles)) {
-                success = false;
-            }
+    for (auto f : systemSpecification.getFunctions()) {
+        if (!writeTestFilesForFunction(*f.second, directory, versionOfTestFiles)) {
+            success = false;
         }
     }
     return success;