Fix screen size issue