diff --git a/tests/test_pipeline_real_weights.py b/tests/test_pipeline_real_weights.py
index 244218e1..1a5a8144 100644
--- a/tests/test_pipeline_real_weights.py
+++ b/tests/test_pipeline_real_weights.py
@@ -9,9 +9,9 @@ import os
 import glob
 
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)) + '/..')
-sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)) + '/../vllm')
 
-from vllm.nvfp4_cutedsl import CuTeDSLMoERunner
+# Direct import from vllm/ subdirectory (not the pip-installed vllm)
+from nvfp4_cutedsl import CuTeDSLMoERunner
 
 # ============================================================
 # CONFIG