fix: adapt to new api