diff --git a/integration/build/build_session_test.go b/integration/build/build_session_test.go
index e1f5ae43ba..51559cbe11 100644
--- a/integration/build/build_session_test.go
+++ b/integration/build/build_session_test.go
@@ -27,7 +27,7 @@ func TestBuildWithSession(t *testing.T) {
 	d.StartWithBusybox(t)
 	defer d.Stop(t)
 
-	client := testEnv.APIClient()
+	client := d.NewClientT(t)
 
 	dockerfile := `
 		FROM busybox
diff --git a/integration/build/build_squash_test.go b/integration/build/build_squash_test.go
index 8acdec596a..8326f2860c 100644
--- a/integration/build/build_squash_test.go
+++ b/integration/build/build_squash_test.go
@@ -26,7 +26,7 @@ func TestBuildSquashParent(t *testing.T) {
 	d.StartWithBusybox(t)
 	defer d.Stop(t)
 
-	client := testEnv.APIClient()
+	client := d.NewClientT(t)
 
 	dockerfile := `
 		FROM busybox