diff --git a/Makefile b/Makefile
index 402119e..be8f30b 100644
--- a/Makefile
+++ b/Makefile
@@ -1,5 +1,5 @@
 # Define the targets
-.PHONY: help venv pip docker-run docs clean
+.PHONY: help venv pip test docker-run docs clean
 
 # Default target
 all: help
@@ -38,6 +38,11 @@ run:
 	@echo "Starting flask server, please wait..."
 	.venv/bin/python ./flask_server.py
 
+# Target to run tests.
+test:
+	@echo "Running tests..."
+	.venv/bin/pytest
+
 # Run entire setup on docker
 docker-run:
 	@docker compose up
diff --git a/pyproject.toml b/pyproject.toml
index 5432643..445fa6b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,3 +5,10 @@ profile = "black"
 ignore = [
     "F841",  # don't complain about unused variables
 ]
+
+[tool.pytest.ini_options]
+minversion = "7.0"
+addopts = "--cov=src"
+pythonpath = [ "src", ]
+testpaths = [ "tests", ]
+
diff --git a/tests/test_class_akku.py b/tests/test_class_akku.py
index bb1a03c..5f958df 100644
--- a/tests/test_class_akku.py
+++ b/tests/test_class_akku.py
@@ -1,6 +1,6 @@
 import unittest
 
-from modules.class_akku import PVAkku
+from akkudoktoreos.class_akku import PVAkku
 
 
 class TestPVAkku(unittest.TestCase):
diff --git a/tests/test_class_ems.py b/tests/test_class_ems.py
index f2bef74..b7979e5 100644
--- a/tests/test_class_ems.py
+++ b/tests/test_class_ems.py
@@ -1,9 +1,9 @@
 import pytest
 
-from modules.class_akku import PVAkku
-from modules.class_ems import EnergieManagementSystem
-from modules.class_haushaltsgeraet import Haushaltsgeraet
-from modules.class_inverter import Wechselrichter  # Example import
+from akkudoktoreos.class_akku import PVAkku
+from akkudoktoreos.class_ems import EnergieManagementSystem
+from akkudoktoreos.class_haushaltsgeraet import Haushaltsgeraet
+from akkudoktoreos.class_inverter import Wechselrichter  # Example import
 
 prediction_hours = 48
 optimization_hours = 24
diff --git a/tests/test_class_optimize.py b/tests/test_class_optimize.py
index fb20dfd..e8ba898 100644
--- a/tests/test_class_optimize.py
+++ b/tests/test_class_optimize.py
@@ -1,7 +1,7 @@
 import numpy as np
 import pytest
 
-from modules.class_optimize import optimization_problem
+from akkudoktoreos.class_optimize import optimization_problem
 
 # Sample known result (replace with the actual expected output)
 EXPECTED_RESULT = {
diff --git a/tests/test_heatpump.py b/tests/test_heatpump.py
index faecdef..e32bfd5 100644
--- a/tests/test_heatpump.py
+++ b/tests/test_heatpump.py
@@ -1,6 +1,6 @@
 import pytest
 
-from modules.heatpump import Heatpump
+from akkudoktoreos.heatpump import Heatpump
 
 
 @pytest.fixture(scope="function")