diff --git a/deeplink_ext/ascend_speed/__init__.py b/deeplink_ext/ascend_speed/__init__.py
index 2a27dbb6..2d380076 100644
--- a/deeplink_ext/ascend_speed/__init__.py
+++ b/deeplink_ext/ascend_speed/__init__.py
@@ -1,3 +1,3 @@
-from .rotary.deeplink import apply_rotary, RotaryEmbedding
+from .rotary_embedding import apply_rotary, RotaryEmbedding
 
 all = ["apply_rotary", "RotaryEmbedding"]
diff --git a/deeplink_ext/ascend_speed/rotary/deeplink.py b/deeplink_ext/ascend_speed/rotary_embedding.py
similarity index 100%
rename from deeplink_ext/ascend_speed/rotary/deeplink.py
rename to deeplink_ext/ascend_speed/rotary_embedding.py