Make sure you assert that the exception is an OperationCanceledException once you change the type.