diff --git a/src/array_api_stubs/_2022_12/indexing_functions.py b/src/array_api_stubs/_2022_12/indexing_functions.py
index e76ce7484..0cbad55ab 100644
--- a/src/array_api_stubs/_2022_12/indexing_functions.py
+++ b/src/array_api_stubs/_2022_12/indexing_functions.py
@@ -1,7 +1,7 @@
-from ._types import Union, array
+from ._types import Union, Optional, array
 
 
-def take(x: array, indices: array, /, *, axis: int) -> array:
+def take(x: array, indices: array, /, *, axis: Optional[int] = None) -> array:
     """
     Returns elements of an array along an axis.
 
diff --git a/src/array_api_stubs/_draft/indexing_functions.py b/src/array_api_stubs/_draft/indexing_functions.py
index d57dc91e5..3b218fdac 100644
--- a/src/array_api_stubs/_draft/indexing_functions.py
+++ b/src/array_api_stubs/_draft/indexing_functions.py
@@ -1,7 +1,7 @@
-from ._types import Union, array
+from ._types import Union, Optional, array
 
 
-def take(x: array, indices: array, /, *, axis: int) -> array:
+def take(x: array, indices: array, /, *, axis: Optional[int] = None) -> array:
     """
     Returns elements of an array along an axis.