diff --git a/keras_hub/src/models/mobilenet/mobilenet_image_classifier_test.py b/keras_hub/src/models/mobilenet/mobilenet_image_classifier_test.py
index 4cfefa1d68..a2011acafe 100644
--- a/keras_hub/src/models/mobilenet/mobilenet_image_classifier_test.py
+++ b/keras_hub/src/models/mobilenet/mobilenet_image_classifier_test.py
@@ -89,7 +89,7 @@ def test_all_presets(self):
                 cls=MobileNetImageClassifier,
                 preset=preset,
                 input_data=self.images,
-                expected_output_shape=(1, 1000),
+                expected_output_shape=(2, 1000),
             )
 
     @pytest.mark.large