Correcting minor UI bug on small screens