diff --git a/include/typings/array2d.pyi b/include/typings/array2d.pyi
index 0a7ea5e1..50feb3d0 100644
--- a/include/typings/array2d.pyi
+++ b/include/typings/array2d.pyi
@@ -15,7 +15,7 @@ class array2d[T]:
     @property
     def numel(self) -> int: ...
 
-    def __new__(cls, n_cols: int, n_rows: int, default: Callable[[vec2i], T] = None): ...
+    def __new__(cls, n_cols: int, n_rows: int, default: T | Callable[[vec2i], T] | None = None): ...
     def __eq__(self, other: object) -> array2d[bool]: ... # type: ignore
     def __ne__(self, other: object) -> array2d[bool]: ... # type: ignore
     def __repr__(self) -> str: ...