fixes to refresh_model_state