diff --git a/.github/workflows/unittests.yaml b/.github/workflows/unittests.yaml
index 8aab627..5f27a3b 100644
--- a/.github/workflows/unittests.yaml
+++ b/.github/workflows/unittests.yaml
@@ -74,7 +74,7 @@ jobs:
       run: |
         pip uninstall luxonis-ml -y
         pip install \
-          "luxonis-ml[data,nn_archive,utils] @ git+https://github.com/luxonis/luxonis-ml.git@${ML_REF}" \
+          "luxonis-ml[data,nn-archive,utils] @ git+https://github.com/luxonis/luxonis-ml.git@${ML_REF}" \
           --upgrade --force-reinstall
 
     - name: Run tests with coverage [Ubuntu]
diff --git a/requirements.txt b/requirements.txt
index 189d2ec..6b50db1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,7 @@ torchvision>=0.10.1
 Pillow>=7.1.2
 PyYAML>=5.3.1
 gcsfs
-luxonis-ml[data,nn_archive,utils]~=0.8.0
+luxonis-ml[data,nn-archive,utils]~=0.8.0
 onnx==1.21.0
 numpy>=1.19.5,<2.1.0
 onnxruntime>=1.20.1