Fix black screen issue