diff --git a/vllm/patches/deepseek_v4.py b/vllm/patches/deepseek_v4.py
index 6db66b6b..3551dc0c 100644
--- a/vllm/patches/deepseek_v4.py
+++ b/vllm/patches/deepseek_v4.py
@@ -208,7 +208,7 @@ class DeepseekV4FP8Config(Fp8Config):
 
 # Staging kernel imported from standalone module (has proper RNE rounding
 # and subnormal handling for UE4M3 — the old embedded copy was broken).
-from staging_kernel import _stage_deepseek_v4_mega_moe_inputs
+from vllm.model_executor.models.staging_kernel import _stage_deepseek_v4_mega_moe_inputs
 
 
 def make_deepseek_v4_expert_params_mapping(