Fix install issue'