diff --git a/integration/kubernetes/confidential/agent_image.bats b/integration/kubernetes/confidential/agent_image.bats
index e72f4978f..f8470e027 100644
--- a/integration/kubernetes/confidential/agent_image.bats
+++ b/integration/kubernetes/confidential/agent_image.bats
@@ -33,15 +33,11 @@ test_tag="[cc][agent][kubernetes][containerd]"
 setup() {
 	remove_test_image "$image_unsigned_protected" || true
 	setup_containerd
-	if [ "${IMAGE_OFFLOAD_TO_GUEST}" == "yes" ]; then
-		configure_containerd_for_nydus_snapshotter "/etc/containerd/config.toml"
-	fi
+	configure_containerd_for_nydus_snapshotter "/etc/containerd/config.toml"
 	restart_containerd
 	reconfigure_kata
-	if [ "${IMAGE_OFFLOAD_TO_GUEST}" == "yes" ]; then
-		switch_image_service_offload off
-		EXPORT_MODE="image_guest_pull" RUNTIMECLASS="$RUNTIMECLASS" configure_nydus_snapshotter
-	fi
+	switch_image_service_offload off
+	EXPORT_MODE="image_guest_pull" RUNTIMECLASS="$RUNTIMECLASS" configure_nydus_snapshotter
 }
 
 @test "$test_tag Test can launch pod with measured boot enabled" {