Switch update_params to prepare