diff --git a/unittests/test_json_schema_exporter.py b/unittests/test_json_schema_exporter.py
index 5768d2d8c0c9acdc9ec355b3f21c4fb5275a3711..99ef153834b6ca558ddf29ca3dbde936421f764e 100644
--- a/unittests/test_json_schema_exporter.py
+++ b/unittests/test_json_schema_exporter.py
@@ -41,13 +41,13 @@ def _mock_execute_query(query_string, unique=False, **kwargs):
     ])
     all_records.extend(other_type_records)
 
-    if query_string == "FIND RECORD 'OtherType'":
+    if query_string == "SELECT name, id FROM RECORD 'OtherType'":
         return other_type_records
     elif query_string == "FIND RECORDTYPE 'OtherType'" and unique is True:
         return other_type_rt
-    elif query_string == "FIND RECORD":
+    elif query_string == "SELECT name, id FROM RECORD":
         return all_records
-    elif query_string == "FIND FILE":
+    elif query_string == "SELECT name, id FROM FILE":
         return all_files
     else:
         return db.Container()