Use original dataset (not temporary) when getting partition key.