Small internal fix