diff --git a/pathwaysutils/experimental/reshard.py b/pathwaysutils/experimental/reshard.py
index a027092..1f156ff 100644
--- a/pathwaysutils/experimental/reshard.py
+++ b/pathwaysutils/experimental/reshard.py
@@ -61,7 +61,7 @@ def ifrt_hlo_sharding(
       result = {
           "devices": {
               "device_ids": [
-                  device.id for device in sharding._addressable_device_assignment  # pylint: disable=protected-access
+                  device.id for device in sharding._device_assignment  # pylint: disable=protected-access
               ]
           },
           "xla_hlo_sharding": (