diff --git a/psl-python/tests/python/deeppsl/test_tensorflow_models.py b/psl-python/tests/python/deeppsl/test_tensorflow_models.py
index 7e39162c9..50d000dfb 100644
--- a/psl-python/tests/python/deeppsl/test_tensorflow_models.py
+++ b/psl-python/tests/python/deeppsl/test_tensorflow_models.py
@@ -69,7 +69,7 @@ def test_sign_model(self):
         post_train_results = sign_model.internal_eval(test_data, options=options)
 
         with tempfile.TemporaryDirectory(suffix = '_TestNeuPSL') as temp_dir:
-            save_path = os.path.join(temp_dir, 'tensorflow_model')
+            save_path = os.path.join(temp_dir, 'tensorflow_model.h5')
             options['save_path'] = save_path
             options['load_path'] = save_path
             sign_model.internal_save(options=options)
diff --git a/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py b/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py
index db28d6187..1144ecb88 100755
--- a/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py
+++ b/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py
@@ -71,7 +71,7 @@ def internal_save(self, options):
         if 'save_path' not in options:
             return {}
 
-        self._model.save(options['save_path'], save_format='tf')
+        self._model.save(options['save_path'])
         return {}
 
     def load(self, options):