1 files changed, 1 insertions, 1 deletions
diff --git a/hw/imc.c b/hw/imc.c
index 10505f6..4b2e948 100644
--- a/hw/imc.c
+++ b/hw/imc.c
@@ -321,7 +321,7 @@ static void disable_unavailable_units(struct dt_node *dev)
 			avl_vec = (0xffULL) << 56;
 	}
 
-	for (i = 0; i < MAX_NEST_UNITS; i++) {
+	for (i = 0; i < ARRAY_SIZE(nest_pmus); i++) {
 		if (!(PPC_BITMASK(i, i) & avl_vec)) {
 			/* Check if the device node exists */
 			target = dt_find_by_name(dev, nest_pmus[i]);