修复class的错误