diff --git a/requirements-dev.txt b/requirements-dev.txt
index b8b82a23..a51cad7e 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -1,3 +1,4 @@
+luxonis-ml[data,tracker]@git+https://github.com/luxonis/luxonis-ml.git@main
 coverage-badge>=1.1.0
 gdown>=4.2.0
 pre-commit>=3.2.1,<4.0.0
diff --git a/requirements.txt b/requirements.txt
index 8bec2286..f276fff8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,6 @@
 blobconverter>=1.4.2
 lightning>=2.4.0
-#luxonis-ml[all]>=0.1.0
-luxonis-ml[all]@git+https://github.com/luxonis/luxonis-ml.git@dev
+luxonis-ml[data,tracker]>=0.4.0
 onnx>=1.12.0
 onnxruntime>=1.13.1
 onnxsim>=0.4.10