We need to take the logic from here and also add it to here. Otherwise we are putting the plan into the cache (via MultiPlanStage::PickBestPlan) with too few works.