diff --git a/deepmd/pd/model/descriptor/__init__.py b/deepmd/pd/model/descriptor/__init__.py
index 8935371c01..8af9582527 100644
--- a/deepmd/pd/model/descriptor/__init__.py
+++ b/deepmd/pd/model/descriptor/__init__.py
@@ -26,11 +26,11 @@
 __all__ = [
     "BaseDescriptor",
     "DescriptorBlock",
+    "DescrptBlockRepformers",
     "DescrptBlockSeA",
     "DescrptBlockSeAtten",
     "DescrptDPA1",
     "DescrptDPA2",
     "DescrptSeA",
     "prod_env_mat",
-    "DescrptBlockRepformers",
 ]