diff --git a/python/aixexplainer.Dockerfile b/python/aixexplainer.Dockerfile
index d44c4361895..3eef7abe3fe 100644
--- a/python/aixexplainer.Dockerfile
+++ b/python/aixexplainer.Dockerfile
@@ -1,12 +1,14 @@
-FROM python:3.7
+FROM python:3.7-slim
 
 COPY third_party third_party
 
 COPY kserve kserve
 RUN pip install --no-cache-dir --upgrade pip && pip install --no-cache-dir -e ./kserve
 
+RUN apt update && apt install -y build-essential
 COPY aixexplainer aixexplainer
 RUN pip install --no-cache-dir -e ./aixexplainer
+RUN apt remove -y build-essential
 
 RUN useradd kserve -m -u 1000 -d /home/kserve
 USER 1000
diff --git a/python/aixexplainer/setup.py b/python/aixexplainer/setup.py
index 9099b85b8a4..0d7ccedee23 100644
--- a/python/aixexplainer/setup.py
+++ b/python/aixexplainer/setup.py
@@ -33,7 +33,7 @@
     install_requires=[
         "kserve>=0.7.0",
         "argparse >= 1.4.0",
-        "aix360 >= 0.1.0",
+        "aix360 >= 0.2.0",
         "lime >= 0.1.1.37",
         "nest_asyncio>=1.4.0",
         "cvxpy == 1.1.13"