Fix screen size bug