fixed the screen issue