Fixes a visualization bug in the class selection.