diff --git a/plat/imx/imx8mm/gpc.c b/plat/imx/imx8mm/gpc.c
index 8b916c4..b2c4832 100644
--- a/plat/imx/imx8mm/gpc.c
+++ b/plat/imx/imx8mm/gpc.c
@@ -474,7 +474,7 @@
 	int i;
 	uint32_t pll_ctrl;
 	/* bypass all the plls before enter DSM mode */
-	for (i = 0; i < 9; i++) {
+	for (i = 0; i < ARRAY_SIZE(pll_ctrl_offset); i++) {
 		pll_ctrl = mmio_read_32(IMX_ANAMIX_BASE + pll_ctrl_offset[i]);
 		pll_ctrl |= PLL_BYPASS;
 		mmio_write_32(IMX_ANAMIX_BASE + pll_ctrl_offset[i], pll_ctrl);
@@ -498,7 +498,7 @@
 	int i;
 	uint32_t pll_ctrl;
 	/* unbypass all the plls after exit from DSM mode */
-	for (i = 0; i < 9; i++) {
+	for (i = 0; i < ARRAY_SIZE(pll_ctrl_offset); i++) {
 		pll_ctrl = mmio_read_32(IMX_ANAMIX_BASE + pll_ctrl_offset[i]);
 		pll_ctrl &= ~PLL_BYPASS;
 		mmio_write_32(IMX_ANAMIX_BASE + pll_ctrl_offset[i], pll_ctrl);