STORM-511 fix updating internal state of PartitionManager