diff --git a/.github/workflows/test_iree.yml b/.github/workflows/test_iree.yml
index 7a7c25173..d4c0bc0e8 100644
--- a/.github/workflows/test_iree.yml
+++ b/.github/workflows/test_iree.yml
@@ -72,6 +72,7 @@ jobs:
     runs-on: nodai-amdgpu-w7900-x86-64
     env:
       VENV_DIR: ${{ github.workspace }}/.venv
+      IREE_TEST_FILES: ~/iree_tests_cache
     steps:
       - name: "Checking out repository"
         uses: actions/checkout@v4
@@ -109,7 +110,6 @@ jobs:
       - name: "Running real weight model tests"
         env:
             IREE_TEST_CONFIG_FILES: iree_tests/configs/config_pytorch_models_cpu_llvm_task.json
-            IREE_TEST_FILES: ~/iree_tests_cache
         run: |
           source ${VENV_DIR}/bin/activate
           pytest iree_tests/pytorch/models -s -n 4 -k real_weights -rpfE --timeout=1200 --retries 2 --retry-delay 5 --durations=0
diff --git a/iree_tests/download_remote_files.py b/iree_tests/download_remote_files.py
index 4b2949f32..427bc3935 100644
--- a/iree_tests/download_remote_files.py
+++ b/iree_tests/download_remote_files.py
@@ -77,10 +77,11 @@ def download_azure_remote_file(test_dir: Path, remote_file: str):
         blob_properties = blob_client.get_blob_properties()
 
         cache_location = os.getenv("IREE_TEST_FILES", default="")
+        print("CACHE LOCATION: " + str(cache_location))
         if cache_location == "":
             os.environ["IREE_TEST_FILES"] = str(REPO_ROOT)
             cache_location = REPO_ROOT
-        if cache_location == REPO_ROOT or remote_file_name.split(".")[1] == "mlirbc":
+        if cache_location == REPO_ROOT:
             local_dir_path = test_dir
             local_file_path = test_dir / remote_file_name
         else: