Yes, these can be replaced by StateTransferManager.isStateTransferInProgressForKey(key). They were left as TODOs because at some point that method was broken. I replaced these now in both tx and no-tx distribution interceptors with DM.isAffectedByRehash(key), which calls StateTransferManager.isStateTransferInProgressForKey(key).