diff --git a/test/e2e/common/node/runtime.go b/test/e2e/common/node/runtime.go
index e455e0c1bb4..ec5bc38e3cb 100644
--- a/test/e2e/common/node/runtime.go
+++ b/test/e2e/common/node/runtime.go
@@ -376,7 +376,7 @@ func ImagePullTest(ctx context.Context, f *framework.Framework, image string, im
 
 	ginkgo.By("check the container status")
 	var latestErr error
-	err := wait.PollUntilContextCancel(ctx, ContainerStatusPollInterval, true, func(ctx context.Context) (bool, error) {
+	err := wait.PollUntilContextTimeout(ctx, ContainerStatusPollInterval, ContainerStatusRetryTimeout, true, func(ctx context.Context) (bool, error) {
 		if latestErr = checkContainerStatus(ctx); latestErr != nil {
 			return false, nil
 		}