2 files changed, 1 insertions, 3 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]);
diff --git a/include/imc.h b/include/imc.h
index 9eaf804..f3d906e 100644
--- a/include/imc.h
+++ b/include/imc.h
@@ -119,8 +119,6 @@ struct imc_chip_cb
 #define NEST_IMC_ENABLE			0x1
 #define NEST_IMC_DISABLE		0x2
 
-#define MAX_NEST_UNITS			48
-
 /*
  * Core IMC SCOMs
  */