BUGFIX: added missing import and corrected function name