When there was only on stage in the CI
config, the white background had only one of the
2 corners rounded which looked weird. This has
been fixed by using a method that determine
which class to apply based on the index.