diff --git a/metropolis/test/e2e/k8s_cts/main.go b/metropolis/test/e2e/k8s_cts/main.go
index 6250a16..b9818b3 100644
--- a/metropolis/test/e2e/k8s_cts/main.go
+++ b/metropolis/test/e2e/k8s_cts/main.go
@@ -105,7 +105,7 @@
 		log.Fatalf("Failed to launch cluster: %v", err)
 	}
 	log.Println("Cluster initialized")
-	clientSet, err := cl.GetKubeClientSet()
+	clientSet, _, err := cl.GetKubeClientSet()
 	if err != nil {
 		log.Fatalf("Failed to get clientSet: %v", err)
 	}
diff --git a/metropolis/test/e2e/suites/kubernetes/run_test.go b/metropolis/test/e2e/suites/kubernetes/run_test.go
index 2baeb17..0c116f6 100644
--- a/metropolis/test/e2e/suites/kubernetes/run_test.go
+++ b/metropolis/test/e2e/suites/kubernetes/run_test.go
@@ -98,7 +98,7 @@
 		t.Fatalf("Could not get curator client: %v", err)
 	}
 	mgmt := apb.NewManagementClient(con)
-	clientSet, err := cluster.GetKubeClientSet()
+	clientSet, _, err := cluster.GetKubeClientSet()
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -281,7 +281,7 @@
 		}
 	}()
 
-	clientSet, err := cluster.GetKubeClientSet()
+	clientSet, _, err := cluster.GetKubeClientSet()
 	if err != nil {
 		t.Fatal(err)
 	}