Fix not found class error