Fix issue on small screens