diff --git a/tests/run_igie.py b/tests/run_igie.py
index 8716fbbb85efa4b66dc94cb8bce1ea01c55700e3..41ef47529dbb9a9b9bf89e72019fb92d17dc9083 100644
--- a/tests/run_igie.py
+++ b/tests/run_igie.py
@@ -598,7 +598,7 @@ def run_nlp_testcase(model, batch_size):
     """
     if model_name == "transformer" and platform.machine() == "aarch64":
         base_script += """
-        export LD_PRELOAD=$(find /usr/local/lib/python3.10/site-packages/scikit_learn.libs -name "libgomp*.so.1.0.0" | head -n1)
+        export LD_PRELOAD=$(find /usr/local/lib/python3.*/site-packages/scikit_learn.libs -name "libgomp*.so.1.0.0" | head -n1)
         export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libGLdispatch.so.0:$LD_PRELOAD
         """
     for prec in model["precisions"]:
diff --git a/tests/run_ixrt.py b/tests/run_ixrt.py
index dfe12410b77ae19bc979dd4b1838fe280cdc6ae8..61a86fad7fbd82ced93b8fd005992e4d9c4c80fb 100644
--- a/tests/run_ixrt.py
+++ b/tests/run_ixrt.py
@@ -642,7 +642,7 @@ def run_nlp_testcase(model, batch_size, whl_url):
                     if model_name == "transformer" and platform.machine() == "aarch64":
                         script = f"""
                         set -x
-                        export LD_PRELOAD=$(find /usr/local/lib/python3.10/site-packages/scikit_learn.libs -name "libgomp*.so.1.0.0" | head -n1)
+                        export LD_PRELOAD=$(find /usr/local/lib/python3.*/site-packages/scikit_learn.libs -name "libgomp*.so.1.0.0" | head -n1)
                         export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libGLdispatch.so.0:$LD_PRELOAD
                         cd ../{model['model_path']}/
                         bash scripts/infer_{model_name}_{prec}_accuracy.sh
@@ -660,7 +660,7 @@ def run_nlp_testcase(model, batch_size, whl_url):
                     if model_name == "transformer" and platform.machine() == "aarch64":
                         script = f"""
                         set -x
-                        export LD_PRELOAD=$(find /usr/local/lib/python3.10/site-packages/scikit_learn.libs -name "libgomp*.so.1.0.0" | head -n1)
+                        export LD_PRELOAD=$(find /usr/local/lib/python3.*/site-packages/scikit_learn.libs -name "libgomp*.so.1.0.0" | head -n1)
                         export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libGLdispatch.so.0:$LD_PRELOAD
                         cd ../{model['model_path']}/
                         bash scripts/infer_{model_name}_{prec}_accuracy.sh --bs {bs}