Fixed layout error on smaller screens