Fixed allgather example