diff --git a/deepmd/descriptor/se_atten.py b/deepmd/descriptor/se_atten.py
index 9c3834e764..ab6f836b41 100644
--- a/deepmd/descriptor/se_atten.py
+++ b/deepmd/descriptor/se_atten.py
@@ -508,7 +508,7 @@ def _compute_dstats_sys_smth(
             sysa2 = [0.0 for i in range(self.ntypes)]
             for ff in range(nframes):
                 natoms = real_natoms_vec[ff]
-                dd_ff = np.reshape(dd_all[ff], [-1, self.ndescrpt * natoms[0]])
+                dd_ff = np.reshape(dd_all[ff], [-1, self.ndescrpt * natoms_vec[0]])
                 start_index = 0
                 for type_i in range(self.ntypes):
                     end_index = (
diff --git a/deepmd/utils/data_system.py b/deepmd/utils/data_system.py
index d87219fcc9..e048dccd99 100644
--- a/deepmd/utils/data_system.py
+++ b/deepmd/utils/data_system.py
@@ -124,6 +124,7 @@ def __init__(
                     rule = int(words[1])
                 self.batch_size = self._make_auto_bs(rule)
             elif "mixed" == words[0]:
+                self.mixed_type = True
                 self.mixed_systems = True
                 if len(words) == 2:
                     rule = int(words[1])