diff --git a/cmd/kube-scheduler/app/options/options_test.go b/cmd/kube-scheduler/app/options/options_test.go index b306d18a60a..c8b93ed98d8 100644 --- a/cmd/kube-scheduler/app/options/options_test.go +++ b/cmd/kube-scheduler/app/options/options_test.go @@ -343,10 +343,8 @@ profiles: // Insulate this test from picking up in-cluster config when run inside a pod // We can't assume we have permissions to write to /var/run/secrets/... from a unit test to mock in-cluster config for testing - originalHost := os.Getenv("KUBERNETES_SERVICE_HOST") - if len(originalHost) > 0 { - os.Setenv("KUBERNETES_SERVICE_HOST", "") - defer os.Setenv("KUBERNETES_SERVICE_HOST", originalHost) + if len(os.Getenv("KUBERNETES_SERVICE_HOST")) > 0 { + t.Setenv("KUBERNETES_SERVICE_HOST", "") } defaultPodInitialBackoffSeconds := int64(1)