diff --git a/test/src/unittests/machinelearning/test_tensorflowpredictmaest.py b/test/src/unittests/machinelearning/test_tensorflowpredictmaest.py
index 572d17a99..819d5456e 100644
--- a/test/src/unittests/machinelearning/test_tensorflowpredictmaest.py
+++ b/test/src/unittests/machinelearning/test_tensorflowpredictmaest.py
@@ -38,8 +38,16 @@ def setUpClass(self):
             testdata.models_dir, "maest", "discogs-maest-10s-pw-1.pb"
         )
 
-        self.model30s = TensorflowPredictMAEST(graphFilename=self.graphFilename30s)
-        self.model10s = TensorflowPredictMAEST(graphFilename=self.graphFilename10s)
+        self.model30s = TensorflowPredictMAEST(
+            graphFilename=self.graphFilename30s,
+            input="serving_default_melspectrogram",
+            output="StatefulPartitionedCall:0",
+        )
+        self.model10s = TensorflowPredictMAEST(
+            graphFilename=self.graphFilename10s,
+            input="serving_default_melspectrogram",
+            output="StatefulPartitionedCall:0",
+        )
 
     @classmethod
     def tearDownClass(self):
diff --git a/test/src/unittests/machinelearning/test_tensorflowpredicttempocnn.py b/test/src/unittests/machinelearning/test_tensorflowpredicttempocnn.py
index f22d1f141..3bed15bc7 100644
--- a/test/src/unittests/machinelearning/test_tensorflowpredicttempocnn.py
+++ b/test/src/unittests/machinelearning/test_tensorflowpredicttempocnn.py
@@ -112,7 +112,7 @@ def testEmptyInput(self):
     def testInvalidParam(self):
         model = join(testdata.models_dir, 'tempocnn', 'deeptemp_k16.pb')
         self.assertConfigureFails(TensorflowPredictTempoCNN(), {'graphFilename': model,
-                                                                'batchSize': 0})
+                                                                'batchSize': -2})
         self.assertConfigureFails(TensorflowPredictTempoCNN(), {'graphFilename': model,
                                                                 'input': 'non_existing'})
         self.assertConfigureFails(TensorflowPredictTempoCNN(), {'graphFilename': model,