fixed the class error
fixed error