diff --git a/tests/setup_gate.sh b/tests/setup_gate.sh
index 2446d4f8eab460de1569e2be34cfae141bcbfdc7..5ace1123659eb5a6e33e02a894bb46334a656ec2 100755
--- a/tests/setup_gate.sh
+++ b/tests/setup_gate.sh
@@ -118,7 +118,7 @@ EOF
     python3 -m venv ~/kolla-venv
     . ~/kolla-venv/bin/activate
 
-    pip install "${KOLLA_SRC_DIR}"
+    pip install "${KOLLA_SRC_DIR}" ${CONTAINER_ENGINE}
 
     sudo ~/kolla-venv/bin/kolla-build