fix: actually use the new function