Fix bugs on small screens