Fix tests

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Mar 10, 2024 · 7a71b57 · 7a71b57
1 parent 1cb8860
commit 7a71b57
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/flytekit/unit/core/image_spec/test_image_spec.py b/tests/flytekit/unit/core/image_spec/test_image_spec.py
@@ -108,7 +108,7 @@ def test_build_existing_image_with_force_push():
     image_spec.exist.return_value = True
     image_spec._is_force_push = True
 
-    ImageBuildEngine.build_image = Mock()
+    ImageBuildEngine._build_image = Mock()
 
     ImageBuildEngine.build(image_spec)
-    ImageBuildEngine.build_image.assert_called_once()
+    ImageBuildEngine._build_image.assert_called_once()