diff --git a/tests/module/module_evaluators_test.py b/tests/module/module_evaluators_test.py
index 2ee1030..7de9627 100644
--- a/tests/module/module_evaluators_test.py
+++ b/tests/module/module_evaluators_test.py
@@ -29,7 +29,7 @@ def setUpClass(cls):
         """
         Prepare TopologyDataFrame once for every test
         """
-        
+
         pdb_file = '{0}/{1}.pdb'.format(filepath, cls.ref['pdb_id'])
         mol_file = '{0}/{1}.mol2'.format(filepath, cls.ref['pdb_id'])
 
diff --git a/tests/module/module_topology_dataframe_test.py b/tests/module/module_topology_dataframe_test.py
index 8e338f7..6bc9099 100644
--- a/tests/module/module_topology_dataframe_test.py
+++ b/tests/module/module_topology_dataframe_test.py
@@ -9,10 +9,11 @@
 import os
 
 from numpy import ndarray
-from test_data import *
 
 from interact.md_system import System
+
 from tests.module.unittest_baseclass import UnittestPythonCompatibility
+from tests.module.test_data import *
 
 currpath = os.path.dirname(__file__)