diff --git a/deepmd/entrypoints/test.py b/deepmd/entrypoints/test.py
index d9ccf392f5..fd0393c914 100644
--- a/deepmd/entrypoints/test.py
+++ b/deepmd/entrypoints/test.py
@@ -124,7 +124,7 @@ def test(
         log.info(f"# testing system : {system}")
 
         # create data class
-        tmap = dp.get_type_map() if isinstance(dp, DeepPot) else None
+        tmap = dp.get_type_map()
         data = DeepmdData(
             system,
             set_prefix="set",
diff --git a/source/tests/pt/test_dp_test.py b/source/tests/pt/test_dp_test.py
index c18c3286f6..0427f2b14a 100644
--- a/source/tests/pt/test_dp_test.py
+++ b/source/tests/pt/test_dp_test.py
@@ -152,6 +152,9 @@ def setUp(self):
         self.config["training"]["training_data"]["systems"] = data_file
         self.config["training"]["validation_data"]["systems"] = data_file
         self.config["model"] = deepcopy(model_property)
+        self.config["model"]["type_map"] = [
+            self.config["model"]["type_map"][i] for i in [1, 0, 3, 2]
+        ]
         self.input_json = "test_dp_test_property.json"
         with open(self.input_json, "w") as fp:
             json.dump(self.config, fp, indent=4)