Solved the missing class issue