diff --git a/onnx_array_api/_command_lines_parser.py b/onnx_array_api/_command_lines_parser.py
index e9b69a2..d1eac62 100644
--- a/onnx_array_api/_command_lines_parser.py
+++ b/onnx_array_api/_command_lines_parser.py
@@ -51,7 +51,7 @@ def get_parser_translate() -> ArgumentParser:
     parser.add_argument(
         "-a",
         "--api",
-        choices=["onnx", "light"],
+        choices=["onnx", "light", "onnx-short", "builder"],
         default="onnx",
         help="API to choose, API from onnx package or light API.",
     )