diff --git a/mlx/random.h b/mlx/random.h
index 0dfdab7a..b6837418 100644
--- a/mlx/random.h
+++ b/mlx/random.h
@@ -89,7 +89,7 @@ inline array uniform(
     const Shape& shape,
     const std::optional<array>& key = std::nullopt,
     StreamOrDevice s = {}) {
-  return uniform(shape, float32, key);
+  return uniform(shape, float32, key, s);
 }
 
 /** Generate samples from the standard normal distribution. */