Return model after update