From 8de9ceb7d6fae16081fa985b4ab2bfebb37d40df Mon Sep 17 00:00:00 2001
From: hwiesmann <github@skywind.eu>
Date: Thu, 1 Jan 2026 01:02:50 +0100
Subject: [PATCH] BUG FIX - Addition of missing parameter in random::uniform
 (#2963)

Co-authored-by: Hartwig Wiesmann <hartwig.wiesmann@skywind.eu>
---
 mlx/random.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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. */