fixed 2 erros