diff --git a/ci/docker/test_image.sh b/ci/docker/test_image.sh
index ab4499a930..68a0013d18 100644
--- a/ci/docker/test_image.sh
+++ b/ci/docker/test_image.sh
@@ -19,7 +19,7 @@ set -euo pipefail
 
 # Install dependencies
 apt-get update
-DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends file bzip2
+DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends file bzip2 gcc
 
 # Download test data
 bash datasets/linear_programming/download_pdlp_test_dataset.sh
@@ -37,7 +37,7 @@ chmod -R a+w "$(pwd)"
 
 cat > /opt/cuopt/test.sh <<EOF
 cd /opt/cuopt/cuopt
-pip install pytest pexpect
+python -m pip install pytest pexpect
 export RAPIDS_DATASET_ROOT_DIR=\$(realpath datasets)
 echo '----------------- CLI TEST START ---------------'
 bash python/libcuopt/libcuopt/tests/test_cli.sh