Fix display bug on small screens