diff --git a/Makefile b/Makefile
index 181eaa0..aba6059 100644
--- a/Makefile
+++ b/Makefile
@@ -1,9 +1,9 @@
 .PHONY: style quality install tensorboard clean fix update-worklog
 
 # make sure to test the local checkout in scripts and not the pre-installed one
-export PYTHONPATH = src
+export PYTHONPATH = src 
 
-check_dirs := . src notebooks scripts
+check_dirs := . src notebooks scripts tests
 
 # Development dependencies
 install:
diff --git a/requirements.txt b/requirements.txt
index 4c616ef..3888064 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,3 +15,4 @@ python-dotenv
 loguru
 gradio
 tensorboard
+pytest
\ No newline at end of file