diff --git a/client/client_image_source_test.go b/client/client_image_source_test.go
index 1c680b445..7ed143d3a 100644
--- a/client/client_image_source_test.go
+++ b/client/client_image_source_test.go
@@ -65,11 +65,11 @@ func testClientGatewayCanceledCredentialsCallbackReturns(t *testing.T, sb integr
 	target, err := url.Parse("http://" + registry)
 	require.NoError(t, err)
 
-	proxy := httputil.NewSingleHostReverseProxy(target)
-	director := proxy.Director
-	proxy.Director = func(req *http.Request) {
-		director(req)
-		req.Host = target.Host
+	proxy := &httputil.ReverseProxy{
+		Rewrite: func(pr *httputil.ProxyRequest) {
+			pr.SetURL(target)
+			pr.Out.Host = target.Host
+		},
 	}
 
 	proxyServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {