diff --git a/tensorflow.dockerfile b/tensorflow.dockerfile
index 479159d..d167453 100644
--- a/tensorflow.dockerfile
+++ b/tensorflow.dockerfile
@@ -22,7 +22,7 @@ sudo apt-get update
 sudo apt install -y software-properties-common
 sudo add-apt-repository -y ppa:deadsnakes/ppa
 sudo apt-get update
-sudo apt-get install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev
+sudo apt-get install -y python${PYTHON_VERSION}-full python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-distutils
 sudo apt-get clean
 sudo rm -rf /var/lib/apt/lists/*
 EOF
@@ -30,8 +30,9 @@ EOF
 WORKDIR /home/rocm/tensorflow
 
 # TODO: simplify dependencies
-# > Ref: https://github.com/ROCm/ROCm/issues/1880
-RUN python${PYTHON_VERSION} -m pip install -U --user pip numpy==${NUMPY_VERSION} wheel packaging requests opt_einsum patchelf
+# > Ref: https://github.com/ROCm/ROCm/issues/1880#issuecomment-1547838227
+# RUN python${PYTHON_VERSION} -m ensurepip --upgrade
+RUN python${PYTHON_VERSION} -m pip install -U --user numpy==${NUMPY_VERSION} wheel packaging requests opt_einsum patchelf
 RUN python${PYTHON_VERSION} -m pip install -U --user keras_preprocessing --no-deps
 
 RUN sed -i "s/python3/python${PYTHON_VERSION}/g" build_rocm_python3