Small API Fix