diff --git a/deepmd/utils/neighbor_stat.py b/deepmd/utils/neighbor_stat.py
index d98d76c58f..982ead3549 100644
--- a/deepmd/utils/neighbor_stat.py
+++ b/deepmd/utils/neighbor_stat.py
@@ -179,7 +179,9 @@ def __init__(
         one_type: bool = False,
     ) -> None:
         """Constructor."""
-        super().__init__(ntypes, rcut, one_type)
+        self.ntypes = ntypes
+        self.rcut = rcut
+        self.mixed_type = one_type
         self.auto_batch_size = AutoBatchSize()
         self.neighbor_stat = NeighborStatOP(ntypes, rcut, not one_type)
         self.place_holders = {}