I assume you meant `sync_arg` and `sync_arg` to be `sync_agg` and `update_counter` ?