From 0b7f7438579cb05385f3e288d26d4338c0b2f55e Mon Sep 17 00:00:00 2001
From: Jeffrey Jiang <jjiang@tenstorrent.com>
Date: Mon, 3 Mar 2025 11:11:10 -0600
Subject: [PATCH] fix pair access

---
 ttnn/cpp/ttnn/tensor/storage.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ttnn/cpp/ttnn/tensor/storage.hpp b/ttnn/cpp/ttnn/tensor/storage.hpp
index f74668f284b..6436916528d 100644
--- a/ttnn/cpp/ttnn/tensor/storage.hpp
+++ b/ttnn/cpp/ttnn/tensor/storage.hpp
@@ -330,7 +330,7 @@ struct MultiDeviceStorage {
         TT_ASSERT(
             buffer_it->device() == device,
             "Mismatch between device derived from buffer and device derived from MultiDeviceStorage.");
-        return *buffer_it;
+        return buffer_it->second;
     }
 
     inline std::shared_ptr<Buffer>& get_buffer_for_device(IDevice* device) {