fix API breakage