Fix an import error