diff --git a/pyproject.toml b/pyproject.toml
index 227b123..45a2a34 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -28,16 +28,16 @@ xinfer = "xinfer.cli:main"
 
 [project.optional-dependencies]
 all = [
-    "xinfer[extra]",
-]
-
-extra = [
-    "rich",
+    "xinfer[transformers]",
+    "xinfer[ultralytics]",
 ]
 
 transformers = [
     "transformers",
-    
+]
+
+ultralytics = [
+    "ultralytics",
 ]