Fixed missing class errors.