Fixed missing class name