diff --git a/src/promptflow/tests/tracing_test/e2etests/test_trace.py b/src/promptflow/tests/tracing_test/e2etests/test_trace.py
index d4fb3235d07..0c9fdbcf95f 100644
--- a/src/promptflow/tests/tracing_test/e2etests/test_trace.py
+++ b/src/promptflow/tests/tracing_test/e2etests/test_trace.py
@@ -4,7 +4,7 @@
 import pytest
 from opentelemetry.trace.status import StatusCode
 
-from promptflow._core.openai_injector import inject_openai_api
+from promptflow.tracing._openai_injector import inject_openai_api
 from promptflow.tracing.contracts.trace import TraceType
 
 from ..utils import execute_function_in_subprocess, prepare_memory_exporter