Fix screen turning off when waking up

Fix screen turning off when waking up