Fix layout issue on small screen phones