diff --git a/build-unit-test-docker.sh b/build-unit-test-docker.sh
index 1c8f25f..40c31d7 100755
--- a/build-unit-test-docker.sh
+++ b/build-unit-test-docker.sh
@@ -195,7 +195,6 @@
     python-pip \
     python-setuptools \
     python-socks \
-    python-jsonschema \
     python3 \
     python3-dev\
     python3-yaml \
@@ -252,7 +251,9 @@
 
 RUN pip install inflection
 RUN pip install pycodestyle
+RUN pip install jsonschema
 RUN pip3 install inflection
+RUN pip3 install jsonschema
 RUN pip3 install meson==0.53.2
 
 FROM openbmc-base as openbmc-lcov