diff --git a/deeplink_ext/ops/rms_norm/internevo_rms_norm.py b/deeplink_ext/ops/rms_norm/internevo_rms_norm.py
index 5b828ce..1135634 100644
--- a/deeplink_ext/ops/rms_norm/internevo_rms_norm.py
+++ b/deeplink_ext/ops/rms_norm/internevo_rms_norm.py
@@ -6,7 +6,7 @@
 if platform_type == PlatformType.TORCH_DIPU:
     # from ._mixed_rms_norm_dipu import MixedFusedRMSNorm
     # Due to the accuracy problem of the npu fused operator, a torch combination is used as an alternative.
-    from .interntrain_rms_norm_fallback import MixedRMSNormTorch as MixedFusedRMSNorm
+    from .internevo_rms_norm_fallback import MixedRMSNormTorch as MixedFusedRMSNorm
 else:
     raise ImportError
 
diff --git a/deeplink_ext/ops/rotary_embedding/interntrain_rotary_embedding.py b/deeplink_ext/ops/rotary_embedding/interntrain_rotary_embedding.py
index 3555e62..61f98c6 100644
--- a/deeplink_ext/ops/rotary_embedding/interntrain_rotary_embedding.py
+++ b/deeplink_ext/ops/rotary_embedding/interntrain_rotary_embedding.py
@@ -4,7 +4,7 @@
 
 platform_type = deeplink_ext_get_platform_type()
 if platform_type == PlatformType.TORCH_DIPU:
-    from ._rotary_embedding_dipu import ApplyRotaryEmb, ApplyRotaryEmbQKV_
+    from .interntrain_rotary_embedding_dipu import ApplyRotaryEmb, ApplyRotaryEmbQKV_
 else:
     raise ImportError