diff --git a/rust/py_test/test_launch_server.py b/rust/py_test/test_launch_server.py
index 13fcc8f4e4c..abcf4cb59e1 100644
--- a/rust/py_test/test_launch_server.py
+++ b/rust/py_test/test_launch_server.py
@@ -90,6 +90,8 @@ def popen_launch_server(
         host,
         "--port",
         port,
+        "--base-gpu-id",
+        "1",
     ]
 
     process = subprocess.Popen(command, stdout=None, stderr=None)