(maybe?) fix the screen turning off