diff --git a/arrayfire/library/array_functions.py b/arrayfire/library/array_functions.py
index e8028e5..90956b3 100644
--- a/arrayfire/library/array_functions.py
+++ b/arrayfire/library/array_functions.py
@@ -954,6 +954,7 @@ def replace(lhs: Array, rhs: Array | int | float, conditional: Array, /) -> None
     wrapper.replace_scalar(lhs.arr, conditional.arr, rhs)
 
 
+@afarray_as_array
 def select(lhs: Array | int | float, rhs: Array | int | float, conditional: Array, /) -> Array:
     """
     Conditionally selects elements from one of two sources (ArrayFire arrays or scalars) based on a condition array.