Fix broken build following change to public API