Fix ui for xs devices