Fixing bug for screen size