Update classical.py

deepmodeling · Nov 22, 2023 · d11ef9d · d11ef9d
1 parent 035a33a
commit d11ef9d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dmff/generators/classical.py b/dmff/generators/classical.py
@@ -319,7 +319,7 @@ def overwrite(self, paramset: ParamSet) -> None:
         angle_theta = paramset[self.name]["angle"]
         angle_k = paramset[self.name]["k"]
         angle_msks = paramset.mask[self.name]["angle"]
-        for nnode, key in enumerate(self.bond_keys):
+        for nnode, key in enumerate(self.angle_keys):
             self.ffinfo["Forces"][self.name]["node"][angle_node_indices[nnode]]["attrib"] = {
             }
             self.ffinfo["Forces"][self.name]["node"][angle_node_indices[nnode]