minor change
pass CancellationToken appropriately