Quick bugfix on class names