diff --git a/xlml/utils/xpk.py b/xlml/utils/xpk.py
index 19f5c6a1b..02709146d 100644
--- a/xlml/utils/xpk.py
+++ b/xlml/utils/xpk.py
@@ -84,6 +84,7 @@ def run_workload(
     cmds = [
         "set -xue",
         f"git clone https://github.com/google/xpk {tmpdir}/xpk",
+        f"cd {tmpdir}/xpk && make install && export PATH=$PATH:$PWD/bin",
     ]
     if accelerator_type == GpuVersion.XPK_H100_MEGA.value:
       workload_create_cmd += " --scheduler=gke.io/topology-aware-auto"