Fix power management