[HELIX-248] More IdealState/RebalancerContext compatibility