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();
diff --git a/lib/psci/psci_suspend.c b/lib/psci/psci_suspend.c
index baa3bd4..05bbe39 100644
--- a/lib/psci/psci_suspend.c
+++ b/lib/psci/psci_suspend.c
@@ -258,9 +258,9 @@
 #endif
 
 	if (is_power_down_state != 0U) {
-		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_wakeup(max_off_lvl);