diff --git a/context.go b/context.go
index d3e4c6d7..ef40bc7b 100644
--- a/context.go
+++ b/context.go
@@ -1221,7 +1221,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: