Fix layout issue for small screens