diff --git a/server/server.go b/server/server.go
index b15b557e39..9ce44517c1 100644
--- a/server/server.go
+++ b/server/server.go
@@ -292,7 +292,7 @@ func (s *ArgoRolloutsServer) WatchRolloutInfos(q *rollout.RolloutInfoListQuery,
 			return
 		}
 	}
-	ctx := context.Background()
+	ctx := ws.Context()
 	rolloutIf := s.Options.RolloutsClientset.ArgoprojV1alpha1().Rollouts(q.GetNamespace())
 
 	kubeInformerFactory := kubeinformers.NewSharedInformerFactoryWithOptions(s.Options.KubeClientset, 0, kubeinformers.WithNamespace(q.Namespace))