Small fixes

chrisjonesBSU · Feb 14, 2020 · a56a3f7 · a56a3f7
1 parent f8315c6
commit a56a3f7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/topology/formats/top.py b/topology/formats/top.py
@@ -4,7 +4,7 @@
 from foyer.smarts import SMARTS
 
 from topology.core.element import element_by_atom_type
-from topology.lib.potential_templates import *
+from topology.lib.potential_templates import LennardJonesPotential
 from topology.utils.compatibility import check_compatibility
 from topology.exceptions import TopologyError
 

diff --git a/topology/utils/compatibility.py b/topology/utils/compatibility.py
@@ -10,7 +10,7 @@ def check_compatibility(topology, accepted_potentials):
             raise EngineIncompatibilityError
 
     for connection_type in topology.connection_types:
-        if not _check_single_potential(atom_type, accepted_potentials):
+        if not _check_single_potential(connection_type, accepted_potentials):
             raise EngineIncompatibilityError