Fix UI on small devices