diff --git a/mlx/ops.cpp b/mlx/ops.cpp
index b58a2836e4..8299320198 100644
--- a/mlx/ops.cpp
+++ b/mlx/ops.cpp
@@ -96,7 +96,7 @@ array indices_or_default(
           << " Got indices with invalid dtype. Indices must be integral.";
       throw std::invalid_argument(msg.str());
     }
-    return astype(indices.value(), uint32);
+    return astype(indices.value(), uint32, s);
   }
 
   Shape shape(x.shape().begin(), x.shape().end() - 2);