(Bugfix?) Compare class names instead of using ==/!=