diff --git a/assets/batch_score/components/driver/tests/e2e/aoai_endpoint_test.py b/assets/batch_score/components/driver/tests/e2e/aoai_endpoint_test.py
index 54f96a59a1..2c8d201869 100644
--- a/assets/batch_score/components/driver/tests/e2e/aoai_endpoint_test.py
+++ b/assets/batch_score/components/driver/tests/e2e/aoai_endpoint_test.py
@@ -14,12 +14,7 @@
 cpu_compute_target = "cpu-cluster"
 source_dir = os.getcwd()
 gated_llm_pipeline_filepath = os.path.join(
-    source_dir,
-    "driver",
-    "tests",
-    "e2e",
-    "prs_pipeline_templates",
-    "base_llm.yml")
+    pytest.source_dir, "tests", "e2e", "prs_pipeline_templates", "base_llm.yml")
 
 RUN_NAME = "batch_score_aoai_endpoint_test"
 JOB_NAME = "gated_batch_score_llm"  # Should be equivalent to base_llm.yml's job name
diff --git a/assets/batch_score/components/driver/tests/e2e/conftest.py b/assets/batch_score/components/driver/tests/e2e/conftest.py
index eabcaa696e..2c95973e24 100644
--- a/assets/batch_score/components/driver/tests/e2e/conftest.py
+++ b/assets/batch_score/components/driver/tests/e2e/conftest.py
@@ -12,7 +12,6 @@
 from azure.ai.ml import MLClient, load_component
 from azure.identity import DefaultAzureCredential
 
-from src.batch_score.common.constants import METADATA_JSON_FILENAME
 from .util import _get_component_name, _set_and_get_component_name_ver, create_copy
 
 
diff --git a/assets/batch_score/components/driver/tests/e2e/mir_endpoint_llm_test.py b/assets/batch_score/components/driver/tests/e2e/mir_endpoint_llm_test.py
index 2564cc59cc..cffab7a817 100644
--- a/assets/batch_score/components/driver/tests/e2e/mir_endpoint_llm_test.py
+++ b/assets/batch_score/components/driver/tests/e2e/mir_endpoint_llm_test.py
@@ -14,12 +14,7 @@
 
 source_dir = os.getcwd()
 gated_llm_pipeline_filepath = os.path.join(
-    source_dir,
-    "driver",
-    "tests",
-    "e2e",
-    "prs_pipeline_templates",
-    "base_llm.yml")
+    pytest.source_dir, "tests", "e2e", "prs_pipeline_templates", "base_llm.yml")
 
 RUN_NAME = "batch_score_mir_endpoint_llm_test"
 JOB_NAME = "gated_batch_score_llm"  # Should be equivalent to base_llm.yml's job name
diff --git a/assets/batch_score/components/driver/tests/e2e/serverless_endpoint_test.py b/assets/batch_score/components/driver/tests/e2e/serverless_endpoint_test.py
index 9c92128e3e..a703e315f9 100644
--- a/assets/batch_score/components/driver/tests/e2e/serverless_endpoint_test.py
+++ b/assets/batch_score/components/driver/tests/e2e/serverless_endpoint_test.py
@@ -13,12 +13,7 @@
 # Common configuration
 source_dir = os.getcwd()
 gated_llm_pipeline_filepath = os.path.join(
-    source_dir,
-    "driver",
-    "tests",
-    "e2e",
-    "prs_pipeline_templates",
-    "base_llm.yml")
+    pytest.source_dir, "tests", "e2e", "prs_pipeline_templates", "base_llm.yml")
 
 RUN_NAME = "batch_score_aoai_endpoint_test"
 JOB_NAME = "gated_batch_score_llm"  # Should be equivalent to base_llm.yml's job name