Fix missing power logic