Fixed missing class