Update usage to match native implementation