diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
index 5858bd8..dc54edd 100644
--- a/.github/workflows/unit_tests.yml
+++ b/.github/workflows/unit_tests.yml
@@ -7,7 +7,7 @@ jobs:
   unit_tests:
     strategy:
       matrix:
-        python-version: [ 3.7, 3.8, 3.9, "3.10" ]
+        python-version: [ 3.8, 3.9, "3.10", "3.11" ]
     timeout-minutes: 15
     runs-on: ubuntu-latest
     steps:
@@ -19,8 +19,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install -e .
-          pip install -r requirements/test_requirements.txt
+          pip install -e .[test]
       - name: Run Tests
         run: |
           pytest tests --doctest-modules --junitxml=tests/diana-test-results.xml
diff --git a/setup.py b/setup.py
index 6f3e1f8..6749f0f 100644
--- a/setup.py
+++ b/setup.py
@@ -90,6 +90,7 @@ def find_resource_files():
     package_data={'neon_diana_utils': find_resource_files()},
     include_package_data=True,
     install_requires=get_requirements("requirements.txt"),
+    extras_require={"test": get_requirements("test_requirements.txt")},
     zip_safe=True,
     classifiers=[
         'Intended Audience :: Developers',