Fix screen black at end