Minor API fix