Small API update