diff --git a/tests/unit_tests.py b/tests/unit_tests.py
index e25bc93..3677651 100644
--- a/tests/unit_tests.py
+++ b/tests/unit_tests.py
@@ -1,7 +1,7 @@
 import glob
 import os
 import subprocess
-from unittest import TestCase
+import unittest
 
 from pathlib import Path
 
@@ -18,10 +18,10 @@ from costs.cost import Cost
 from costs.constants import SKIN_RETROFIT, SKIN_RETROFIT_AND_SYSTEM_RETROFIT_AND_PV, SYSTEM_RETROFIT_AND_PV
 
 
-class UnitTests(TestCase):
+class UnitTests(unittest.TestCase):
   def setUp(self) -> None:
-    city_file = "./data/test.geojson"
-    output_path = Path('./output/').resolve()
+    city_file = Path("./tests/data/test.geojson").resolve()
+    output_path = Path('./tests/output/').resolve()
     city = GeometryFactory('geojson',
                            city_file,
                            height_field='citygml_me',