diff --git a/mediapipe/model_maker/python/text/text_classifier/text_classifier_test.py b/mediapipe/model_maker/python/text/text_classifier/text_classifier_test.py
index 7a30d19f..d2edb78b 100644
--- a/mediapipe/model_maker/python/text/text_classifier/text_classifier_test.py
+++ b/mediapipe/model_maker/python/text/text_classifier/text_classifier_test.py
@@ -72,8 +72,10 @@ class TextClassifierTest(tf.test.TestCase):
     self.assertTrue(os.path.exists(output_metadata_file))
     self.assertGreater(os.path.getsize(output_metadata_file), 0)
     self.assertTrue(
-        filecmp.cmp(output_metadata_file,
-                    self._AVERAGE_WORD_EMBEDDING_JSON_FILE))
+        filecmp.cmp(
+            output_metadata_file,
+            self._AVERAGE_WORD_EMBEDDING_JSON_FILE,
+            shallow=False))
 
   def test_create_and_train_bert(self):
     train_data, validation_data = self._get_data()
diff --git a/mediapipe/model_maker/python/vision/image_classifier/image_classifier_test.py b/mediapipe/model_maker/python/vision/image_classifier/image_classifier_test.py
index 6ca21d33..14c67d83 100644
--- a/mediapipe/model_maker/python/vision/image_classifier/image_classifier_test.py
+++ b/mediapipe/model_maker/python/vision/image_classifier/image_classifier_test.py
@@ -135,7 +135,9 @@ class ImageClassifierTest(tf.test.TestCase, parameterized.TestCase):
 
     self.assertTrue(os.path.exists(output_metadata_file))
     self.assertGreater(os.path.getsize(output_metadata_file), 0)
-    self.assertTrue(filecmp.cmp(output_metadata_file, expected_metadata_file))
+    self.assertTrue(
+        filecmp.cmp(
+            output_metadata_file, expected_metadata_file, shallow=False))
 
   def test_continual_training_by_loading_checkpoint(self):
     mock_stdout = io.StringIO()