diff --git a/tools/orchestration/multihost_runner.py b/tools/orchestration/multihost_runner.py
index 8b7205884..1cb708414 100644
--- a/tools/orchestration/multihost_runner.py
+++ b/tools/orchestration/multihost_runner.py
@@ -172,6 +172,7 @@ def get_slices():
       "gcloud",
       "compute",
       "tpus",
+      "tpu-vm",
       "describe",
       slice_names[0],
       "--flatten=networkEndpoints[]",