Fix screen in/out transitions