API fix.
Small fix.