From 605c6bc55130d6fed61d6bc4e1270b5e5bea2aff Mon Sep 17 00:00:00 2001
From: Hyunsung Lee <ita9naiwa@gmail.com>
Date: Tue, 22 Apr 2025 05:55:41 +0900
Subject: [PATCH] nit

---
 python/src/ops.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/src/ops.cpp b/python/src/ops.cpp
index 2572a6d53..5969c5052 100644
--- a/python/src/ops.cpp
+++ b/python/src/ops.cpp
@@ -5196,8 +5196,8 @@ void init_ops(nb::module_& m) {
           throw std::invalid_argument(
               "[broadcast_shapes] Must provide at least one shape.");
 
-        mx::Shape result;
-        for (size_t i = 0; i < shapes.size(); ++i) {
+        mx::Shape result = nb::cast<mx::Shape>(shapes[0]);
+        for (size_t i = 1; i < shapes.size(); ++i) {
           if (!nb::isinstance<mx::Shape>(shapes[i]) &&
               !nb::isinstance<nb::tuple>(shapes[i]))
             throw std::invalid_argument(