Small internal API change