diff --git a/tests/integ/test_tfs.py b/tests/integ/test_tfs.py
index a53ed52c2a..8d8f8083ef 100644
--- a/tests/integ/test_tfs.py
+++ b/tests/integ/test_tfs.py
@@ -171,6 +171,7 @@ def test_predict_with_accelerator(tfs_predictor_with_accelerator):
     assert expected_result == result
 
 
+@pytest.mark.local_mode
 def test_predict_with_entry_point(tfs_predictor_with_model_and_entry_point_same_tar):
     input_data = {"instances": [1.0, 2.0, 5.0]}
     expected_result = {"predictions": [4.0, 4.5, 6.0]}
@@ -179,6 +180,7 @@ def test_predict_with_entry_point(tfs_predictor_with_model_and_entry_point_same_
     assert expected_result == result
 
 
+@pytest.mark.local_mode
 def test_predict_with_model_and_entry_point_and_dependencies_separated(
     tfs_predictor_with_model_and_entry_point_and_dependencies
 ):