Fixed CancellationToken

Passing all tests