fixing an import error