diff --git a/.github/workflows/unit-tests.yml b/.github/workflows/unit-tests.yml
index 9d1f054..8bfc776 100644
--- a/.github/workflows/unit-tests.yml
+++ b/.github/workflows/unit-tests.yml
@@ -3,11 +3,11 @@ name: Test package
 on:
   push:
     branches:
-      - main
+      - master
   
   pull_request:
     branches:
-      - main
+      - master
     
   workflow_dispatch:
   workflow_call:
diff --git a/pyproject.toml b/pyproject.toml
index 60636da..760c9f6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -15,7 +15,7 @@ dependencies = ["numpy"]
 
 
 [project.optional-dependencies]
-
+test = ["pytest"]
 [tool.pytest.ini_options]
 addopts = ["--import-mode=importlib"]
 testpaths = ["tests"]