From d12a41e6f7d8011ccc28cd446a08b00c869a3b90 Mon Sep 17 00:00:00 2001
From: Sadi Kneipp <sadi.neto13@gmail.com>
Date: Tue, 3 Dec 2024 03:27:58 +0000
Subject: [PATCH] fix typo

---
 pathwaysutils/persistence/pathways_orbax_handler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pathwaysutils/persistence/pathways_orbax_handler.py b/pathwaysutils/persistence/pathways_orbax_handler.py
index 4148210..0bc6db9 100644
--- a/pathwaysutils/persistence/pathways_orbax_handler.py
+++ b/pathwaysutils/persistence/pathways_orbax_handler.py
@@ -155,7 +155,7 @@ async def deserialize(
             grouped_shardings = [shardings[idx] for idx in idxs]
             locations, names = extract_parent_dir_and_name(grouped_infos)
             grouped_arrays = helper.read_arrays(
-                location[0],
+                locations[0],
                 names,
                 grouped_dtypes,
                 grouped_global_shapes,