[FIX]: small fix for new api