diff --git a/build.py b/build.py
index 150719ccc0..04e1021f26 100755
--- a/build.py
+++ b/build.py
@@ -1459,15 +1459,15 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
 # python3, python3-pip and some pip installs required for the python backend
 RUN apt-get update \\
       && apt-get install -y --no-install-recommends \\
-            python3 \\
             libarchive-dev \\
-            python3-pip \\
-            python3-wheel \\
-            python3-setuptools \\
             libpython3-dev \\
+            python3 \\
+            python3-pip \\
       && pip3 install --upgrade \\
             \"numpy<2\" \\
             virtualenv \\
+            setuptools \\
+            wheel \\
       && rm -rf /var/lib/apt/lists/*
 """
     if "tensorrtllm" in backends: