fix errors on classes