diff --git a/deepmd/__init__.py b/deepmd/__init__.py
index 64386b754c..3a295dcbef 100644
--- a/deepmd/__init__.py
+++ b/deepmd/__init__.py
@@ -4,7 +4,7 @@
 
 from . import cluster, descriptor, fit, loss, utils
 from .env import set_mkl
-from .infer import DeepPotential
+from .infer import DeepEval, DeepPotential
 from .infer.data_modifier import DipoleChargeModifier
 
 set_mkl()
diff --git a/deepmd/entrypoints/__init__.py b/deepmd/entrypoints/__init__.py
index d92cbf3076..80d0431f8c 100644
--- a/deepmd/entrypoints/__init__.py
+++ b/deepmd/entrypoints/__init__.py
@@ -17,7 +17,7 @@
     "doc_train_input",
     "freeze",
     "test",
-    "train",
+    "train_dp",
     "transfer",
     "compress",
     "doc_train_input",