Fixed missing class error