Fix screen stuck issue