diff --git a/deepmd/pt/model/model/__init__.py b/deepmd/pt/model/model/__init__.py
index f3896ec853..03896f3e59 100644
--- a/deepmd/pt/model/model/__init__.py
+++ b/deepmd/pt/model/model/__init__.py
@@ -180,8 +180,8 @@ def get_zbl_model(model_params):
     filepath = model_params["use_srtab"]
     pt_model = PairTabAtomicModel(
         filepath,
-        model_params["descriptor"]["rcut"],
-        model_params["descriptor"]["sel"],
+        descriptor.get_rcut(),
+        descriptor.get_sel(),
         type_map=model_params["type_map"],
     )