diff --git a/deepmd/model/model.py b/deepmd/model/model.py
index 9ae5eacf4f..884a95cc46 100644
--- a/deepmd/model/model.py
+++ b/deepmd/model/model.py
@@ -531,7 +531,7 @@ def __init__(
             self.descrpt = descriptor
         else:
             self.descrpt = Descriptor(
-                **descriptor, ntypes=len(type_map), spin=self.spin
+                **descriptor, ntypes=len(self.type_map), spin=self.spin
             )
 
         if isinstance(fitting_net, Fitting):