Implement code suggestions

openmm · Feb 27, 2025 · f918615 · f918615
1 parent d3f529e
commit f918615
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openmmforcefields/generators/template_generators.py b/openmmforcefields/generators/template_generators.py
@@ -1592,9 +1592,9 @@ def generate_residue_template(self, molecule, residue_atoms=None):
 
         # Remove CMMotionRemover if present
         # See https://github.com/openmm/openmmforcefields/issues/365
-        for f_idx in range(system.getNumForces()):
+        for f_idx in reversed(range(system.getNumForces())):
             force = system.getForce(f_idx)
-            if "CMMotionRemover" in str(force):
+            if isinstance(force, CMMotionRemover):
                 system.removeForce(f_idx)
 
         self.cache_system(smiles, system)