diff --git a/deepmd/tf/loss/tensor.py b/deepmd/tf/loss/tensor.py
index d3a1b95369..4a70ae2a96 100644
--- a/deepmd/tf/loss/tensor.py
+++ b/deepmd/tf/loss/tensor.py
@@ -148,7 +148,7 @@ def label_requirement(self) -> List[DataRequirementItem]:
         # data required
         data_requirements.append(
             DataRequirementItem(
-                "atomic_" + self.label_name,
+                "atom_" + self.label_name,
                 self.tensor_size,
                 atomic=True,
                 must=False,