Fixing bug that caused the class icon to repeat