Fixing an import error