fix errors

thx shani