diff --git a/integration-cli/check_test.go b/integration-cli/check_test.go
index 87517e7db7..3ee1a56245 100644
--- a/integration-cli/check_test.go
+++ b/integration-cli/check_test.go
@@ -67,7 +67,7 @@ func TestMain(m *testing.M) {
 func Test(t *testing.T) {
 	cli.SetTestEnvironment(testEnv)
 	fakestorage.SetTestEnvironment(&testEnv.Execution)
-	ienv.ProtectImages(t, &testEnv.Execution)
+	ienv.ProtectAll(t, &testEnv.Execution)
 	check.TestingT(t)
 }
 
diff --git a/integration/container/main_test.go b/integration/container/main_test.go
index 1c4e078400..fbfed2ae40 100644
--- a/integration/container/main_test.go
+++ b/integration/container/main_test.go
@@ -23,6 +23,6 @@ func TestMain(m *testing.M) {
 }
 
 func setupTest(t *testing.T) func() {
-	environment.ProtectImages(t, testEnv)
+	environment.ProtectAll(t, testEnv)
 	return func() { testEnv.Clean(t) }
 }
diff --git a/integration/service/main_test.go b/integration/service/main_test.go
index 4d6af81895..4cad6ed975 100644
--- a/integration/service/main_test.go
+++ b/integration/service/main_test.go
@@ -25,6 +25,6 @@ func TestMain(m *testing.M) {
 }
 
 func setupTest(t *testing.T) func() {
-	environment.ProtectImages(t, testEnv)
+	environment.ProtectAll(t, testEnv)
 	return func() { testEnv.Clean(t) }
 }