fixes for update feature