Fixed missing class issues.