Small bug fixes

update on update methode