StateTransferManager.isStateTransferInProgressForKey(key) is not used in all places where it should. For example TxDistributionIntercetptor.isStateTransferInProgressForKey() has an ad-hoc implementation.

There are also other places, some are marked with TODOs, in the state transfer and distribution area. We need to locate all of them and upgrade this code.