fix errors
fix to work