Fix operator api in distributed sample