diff --git a/tests/s2_inference/test_vectorise_inference_cache.py b/tests/s2_inference/test_vectorise_inference_cache.py
index 82342414d..10748a982 100644
--- a/tests/s2_inference/test_vectorise_inference_cache.py
+++ b/tests/s2_inference/test_vectorise_inference_cache.py
@@ -1,9 +1,11 @@
+import importlib
 import os
 import random
+import sys
 import unittest.mock
+from concurrent.futures import ThreadPoolExecutor
 from unittest.mock import patch
 
-from concurrent.futures import ThreadPoolExecutor
 import numpy as np
 from PIL import Image
 
@@ -14,6 +16,9 @@ def _import_vectorise_with_inference_cache(self, cache_size: int = 50, cache_typ
         """Import the vectorise function with the specified cache size and type."""
         os.environ["MARQO_INFERENCE_CACHE_TYPE"] = cache_type
         os.environ["MARQO_INFERENCE_CACHE_SIZE"] = str(cache_size)
+
+        # Reload the module to apply the new environment variables
+        importlib.reload(sys.modules['marqo.s2_inference.s2_inference'])
         from marqo.s2_inference.s2_inference import vectorise
         return vectorise