Move power state change to go