diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index df6c846295ab4d461d383981cd91452aafe148f5..83c2a8e7978f5b2f7c9fd5075f0cfbf5f8e19d45 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -36,4 +36,4 @@ unit-test-job:   # This job runs in the test stage.
     - python3 --version
     - pip install -r requirements.txt
     - sh install.sh
-    - python3 tests/test_flip_image.py
+    - pytest
diff --git a/requirements.txt b/requirements.txt
index 5afcfb3fa2b751e208cd07b8b0d351a86bb2873d..66fd8a62f0a5bdfa49dbfdde997d70b8d0949b53 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,4 +2,5 @@ numpy~=1.20.3
 joblib~=1.1.0
 #csst~=0.0.1
 setuptools~=58.0.4
-astropy~=5.1
\ No newline at end of file
+astropy~=5.1
+pytest
\ No newline at end of file