fix code
fixes