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,