[FIX] Correct _update_models