#define to_scmi_pd(gpd) container_of(gpd, struct scmi_pm_domain, genpd)
 
-static int scmi_pd_power(struct generic_pm_domain *domain, bool power_on)
+static int scmi_pd_power(struct generic_pm_domain *domain, u32 state)
 {
-       u32 state;
        struct scmi_pm_domain *pd = to_scmi_pd(domain);
 
-       if (power_on)
-               state = SCMI_POWER_STATE_GENERIC_ON;
-       else
-               state = SCMI_POWER_STATE_GENERIC_OFF;
-
        return power_ops->state_set(pd->ph, pd->domain, state);
 }
 
 static int scmi_pd_power_on(struct generic_pm_domain *domain)
 {
-       return scmi_pd_power(domain, true);
+       return scmi_pd_power(domain, SCMI_POWER_STATE_GENERIC_ON);
 }
 
 static int scmi_pd_power_off(struct generic_pm_domain *domain)
 {
-       return scmi_pd_power(domain, false);
+       return scmi_pd_power(domain, SCMI_POWER_STATE_GENERIC_OFF);
 }
 
 static int scmi_pm_domain_probe(struct scmi_device *sdev)