diff --git a/TestMain.py b/TestMain.py
index 72af55c..e06555c 100644
--- a/TestMain.py
+++ b/TestMain.py
@@ -61,7 +61,8 @@ def test_build_specimin_command(self):
         targets = [{
                     "method": "getMode(ColumnMetadata, Map<String, String>)",
                     "file": "IndexMode.java",
-                    "package": 'org.apache.cassandra.index.sasi.conf'
+                    "package": 'org.apache.cassandra.index.sasi.conf',
+                    "model": "cf"
                    }]
  
         target_dir = '/user/ISSUES/cf-6077'
@@ -76,7 +77,8 @@ def test_build_specimin_command(self):
         targets = [{
                     "method": "transform(String, byte[])",
                     "file": "Avro2Confluent.java",
-                    "package": 'com.fillmore_labs.kafka.sensors.serde.confluent.interop'
+                    "package": 'com.fillmore_labs.kafka.sensors.serde.confluent.interop',
+                    "model" : "cf"
                    }]
 
         target_dir = '/user/ISSUES/cf-6019'
@@ -109,7 +111,8 @@ def test_run_specimin(self):
         targets = [{
                     "method": "bar()",
                     "file": "Simple.java",
-                    "package": "com.example"
+                    "package": "com.example",
+                    "model" : "cf"
                    }]
         target_dir = 'resources/onefilesimple'
 
@@ -121,4 +124,4 @@ def test_run_specimin(self):
 
 
 if __name__ == '__main__':
-    unittest.main()
\ No newline at end of file
+    unittest.main()