fix update model unit tests