Fix updating of models