API fix

Signed-off-by: Nigel Magnay <nigel.magnay@gmail.com>