fix update behaviour