diff --git a/examples/custom_workflows/inference_analysis/build_docker_sglang.sh b/examples/custom_workflows/inference_analysis/build_docker_sglang.sh
index 4c0b98e7..7d0c9f24 100755
--- a/examples/custom_workflows/inference_analysis/build_docker_sglang.sh
+++ b/examples/custom_workflows/inference_analysis/build_docker_sglang.sh
@@ -207,7 +207,7 @@ RUN SGLANG_DIR=\$(pip show sglang | grep "Editable project location" | cut -d' '
             { echo "Failed to apply \$patch"; exit 1; }; \\
         fi \\
     done && \\
-    pip install --no-deps /tmp/TraceLens && \\
+    pip install --upgrade /tmp/TraceLens && \\
     rm -rf /tmp/TraceLens
 
 WORKDIR /workspace
diff --git a/examples/custom_workflows/inference_analysis/build_docker_vllm.sh b/examples/custom_workflows/inference_analysis/build_docker_vllm.sh
index 9f137c67..570df4ce 100644
--- a/examples/custom_workflows/inference_analysis/build_docker_vllm.sh
+++ b/examples/custom_workflows/inference_analysis/build_docker_vllm.sh
@@ -122,7 +122,7 @@ COPY . /tmp/TraceLens
 RUN VLLM_DIR=\$(python -c "import vllm, os; print(os.path.join(os.path.dirname(vllm.__file__), '..'))") && \\
     cd "\${VLLM_DIR}" && \\
     (git apply /tmp/TraceLens/${PATCH_PATH} || patch -p1 --fuzz=10 < /tmp/TraceLens/${PATCH_PATH}) && \\
-    pip install --no-deps /tmp/TraceLens && \\
+    pip install --upgrade /tmp/TraceLens && \\
     rm -rf /tmp/TraceLens
 
 WORKDIR /workspace