diff --git a/tests/setup_gate.sh b/tests/setup_gate.sh
index 2446d4f8ea..5ace112365 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