Fixed styling to make it consistent across editing screens