The 'cancel' flag in the PartitionCountingCallback doesn't need to be an AtomicBoolean