diff --git a/tests/__init__.py b/tests/__init__.py
deleted file mode 100644
index e69de29..0000000
diff --git a/tests/generate_component_data.py b/tests/generate_component_data.py
index 1c19109..24e160b 100644
--- a/tests/generate_component_data.py
+++ b/tests/generate_component_data.py
@@ -1,5 +1,5 @@
-from common import mol_h2o_3, specifications
-from utils import generate_component_data
+from .common import mol_h2o_3, specifications
+from .utils import generate_component_data
 
 # Generates test data from component
 # This does not generate reference data, but the outputs of the individual calculations