diff --git a/test/sagemaker_tests/tensorflow/inference/test/data/python-with-requirements_greater_than_equal_to_26.tar.gz b/test/sagemaker_tests/tensorflow/inference/test/data/python-with-requirements_greater_than_equal_to_26.tar.gz
new file mode 100644
index 000000000000..beeb7c085558
Binary files /dev/null and b/test/sagemaker_tests/tensorflow/inference/test/data/python-with-requirements_greater_than_equal_to_26.tar.gz differ
diff --git a/test/sagemaker_tests/tensorflow/inference/test/data/tfs-model_greater_than_equal_to_tf26.tar.gz b/test/sagemaker_tests/tensorflow/inference/test/data/tfs-model_greater_than_equal_to_tf26.tar.gz
new file mode 100644
index 000000000000..bd7e0b24ecfe
Binary files /dev/null and b/test/sagemaker_tests/tensorflow/inference/test/data/tfs-model_greater_than_equal_to_tf26.tar.gz differ
diff --git a/test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/test_tfs.py b/test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/test_tfs.py
index a0fbfd284cd9..81d2d7be4c15 100644
--- a/test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/test_tfs.py
+++ b/test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/test_tfs.py
@@ -18,9 +18,9 @@
 from ...... import invoke_sm_endpoint_helper_function
 
 
-MODEL_WITH_REQUIREMENTS_PATH = "data/tfs-model_greater_than_equal_to_tf26.tar.gz"
-MODEL_WITH_LIB_PATH = "data/python-with-lib.tar.gz"
-TFS_MODEL_PATH = "data/tfs-model.tar.gz"
+MODEL_WITH_REQUIREMENTS_PATH = "test/data/tfs-model_greater_than_equal_to_tf26.tar.gz"
+MODEL_WITH_LIB_PATH = "test/data/python-with-lib.tar.gz"
+TFS_MODEL_PATH = "test/data/tfs-model.tar.gz"
 TFS_NEURONX_MODEL_PATH = "data/tfs-neuronx-model.tar.gz"
 TFS_NEURON_MODEL_PATH = "data/tfs-neuron-model.tar.gz"
 MME1_MODEL_PATHS = [
diff --git a/test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/util.py b/test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/util.py
index a259252a3236..0cf25b35752f 100644
--- a/test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/util.py
+++ b/test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/util.py
@@ -20,10 +20,12 @@
 import time
 import sys
 
+from test.test_utils import LOGGER
+
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)
 logger.addHandler(logging.StreamHandler(sys.stdout))
-BATCH_CSV = os.path.join("data", "batch.csv")
+BATCH_CSV = os.path.join("test", "data", "batch.csv")
 
 
 def _botocore_resolver():
@@ -174,7 +176,6 @@ def find_or_put_mme_model_data(region, boto_session, mme_folder_name, path_list)
         except botocore.exceptions.ClientError as e:
             if e.response["Error"]["Code"] != "404":
                 raise
-
             # file doesn't exist - upload it
             s3.upload_file(local_path, bucket, key)