diff --git a/lib/psci/psci_off.c b/lib/psci/psci_off.c
index dbc646c..577fdd7 100644
--- a/lib/psci/psci_off.c
+++ b/lib/psci/psci_off.c
@@ -163,9 +163,9 @@
 		    RT_INSTR_ENTER_HW_LOW_PWR,
 		    PMF_NO_CACHE_MAINT);
 #endif
-		if (psci_plat_pm_ops->pwr_domain_pwr_down_wfi != NULL) {
+		if (psci_plat_pm_ops->pwr_domain_pwr_down != NULL) {
 			/* This function may not return */
-			psci_plat_pm_ops->pwr_domain_pwr_down_wfi(&state_info);
+			psci_plat_pm_ops->pwr_domain_pwr_down(&state_info);
 		}
 
 		psci_pwrdown_cpu_end_terminal();