diff --git a/deepmd_utils/model_format/network.py b/deepmd_utils/model_format/network.py
index aec62d74af..c73e53f5cb 100644
--- a/deepmd_utils/model_format/network.py
+++ b/deepmd_utils/model_format/network.py
@@ -512,7 +512,7 @@ def serialize(self) -> dict:
         }
 
     @classmethod
-    def deserialize(cls, data: dict) -> "EmbeddingNet":
+    def deserialize(cls, data: dict) -> "FittingNet":
         """Deserialize the network from a dict.
 
         Parameters