diff --git a/context.go b/context.go
index d5ef8b81..0323478c 100644
--- a/context.go
+++ b/context.go
@@ -1260,7 +1260,7 @@ func (c *Context) SSEvent(name string, message any) {
 // indicates "Is client disconnected in middle of stream"
 func (c *Context) Stream(step func(w io.Writer) bool) bool {
 	w := c.Writer
-	clientGone := w.CloseNotify()
+	clientGone := c.Request.Context().Done()
 	for {
 		select {
 		case <-clientGone: