fix: add imports to the functions