ARM: PSCI: Rework the DT handler slightly