diff --git a/api/notification/live.go b/api/notification/live.go
index 41a32b56..d5ab32b0 100644
--- a/api/notification/live.go
+++ b/api/notification/live.go
@@ -1,12 +1,14 @@
 package notification
 
 import (
+	"io"
+	"time"
+
 	"github.com/0xJacky/Nginx-UI/api"
+	"github.com/0xJacky/Nginx-UI/internal/kernel"
 	"github.com/0xJacky/Nginx-UI/internal/notification"
 	"github.com/0xJacky/Nginx-UI/model"
 	"github.com/gin-gonic/gin"
-	"io"
-	"time"
 )
 
 func Live(c *gin.Context) {
@@ -38,6 +40,8 @@ func Live(c *gin.Context) {
 		case <-notify:
 			notification.RemoveClient(c)
 			return
+		case <-kernel.Context.Done():
+			return
 		}
 	}
 }