fix: Match the new api