diff --git a/jigsawstack/vision.py b/jigsawstack/vision.py
index 280b71d..8eccb51 100644
--- a/jigsawstack/vision.py
+++ b/jigsawstack/vision.py
@@ -102,9 +102,9 @@ class ObjectDetectionParams(TypedDict):
     List of prompts for object detection
     """
 
-    features: NotRequired[List[Literal["object_detection", "gui"]]]
+    features: NotRequired[List[Literal["object", "gui"]]]
     """
-    List of features to enable: object_detection, gui
+    List of features to enable: object, gui
     """
 
     annotated_image: NotRequired[bool]
diff --git a/tests/test_object_detection.py b/tests/test_object_detection.py
index 1fbd5ca..b7286aa 100644
--- a/tests/test_object_detection.py
+++ b/tests/test_object_detection.py
@@ -49,7 +49,7 @@
         "name": "with_blob_both_features",
         "blob": IMAGE_URL,
         "options": {
-            "features": ["object_detection", "gui"],
+            "features": ["object", "gui"],
             "annotated_image": True,
             "return_type": "url",
         },
@@ -63,7 +63,7 @@
         "name": "with_blob_object_detection_features",
         "blob": IMAGE_URL,
         "options": {
-            "features": ["object_detection"],
+            "features": ["object"],
             "annotated_image": True,
             "return_type": "base64",
         },
@@ -80,7 +80,7 @@
         "name": "with_all_options",
         "blob": IMAGE_URL,
         "options": {
-            "features": ["object_detection", "gui"],
+            "features": ["object", "gui"],
             "prompts": ["car", "road", "tree"],
             "annotated_image": True,
             "return_type": "base64",