diff --git a/python/test/test_api.py b/python/test/test_api.py
index 452429ed9..c15847aab 100644
--- a/python/test/test_api.py
+++ b/python/test/test_api.py
@@ -357,8 +357,11 @@ def test_stop(self):
                     {
                         "backend": "python",
                         "parameters": {"decoupled": {"string_value": "False"}},
-                        # Keep instance count low for fast startup/cleanup
-                        "instance_group": [{"kind": "KIND_GPU", "count": 1}],
+                        # Keep instance count low for fast startup/cleanup.
+                        # Alternatively can use KIND_CPU here, but keeping gpus/count explicit.
+                        "instance_group": [
+                            {"kind": "KIND_GPU", "gpus": [0], "count": 1}
+                        ],
                     }
                 )
             },