Fix screen on problem