diff --git a/deepmd/train/trainer.py b/deepmd/train/trainer.py
index e301a0753b..72dc61418e 100644
--- a/deepmd/train/trainer.py
+++ b/deepmd/train/trainer.py
@@ -315,7 +315,7 @@ def build (self,
                 self._init_from_frz_model()
             
             self.neighbor_stat \
-                = NeighborStat(self.ntypes, self.descrpt_param['rcut'])
+                = NeighborStat(self.ntypes, self.descrpt.get_rcut())
             self.min_nbor_dist, self.max_nbor_size \
                 = self.neighbor_stat.get_stat(data)
             tf.constant(self.min_nbor_dist,