diff --git a/VDS/Components/Notification/Notification.swift b/VDS/Components/Notification/Notification.swift
index 527ff307..19dca1c7 100644
--- a/VDS/Components/Notification/Notification.swift
+++ b/VDS/Components/Notification/Notification.swift
@@ -159,11 +159,10 @@ public class Notification: View {
             $0.size = .small
         }
         
-        firstButton.publisher(for: .touchUpInside).sink { button in
+        firstButton.onClickSubscriber = firstButton.publisher(for: .touchUpInside).sink { button in
             print("\(button.text) has been pressed")
-        }.store(in: &subscribers)
-        
-        
+        }
+                
         #warning("Upon adding the button into the stack view, button is visible. Where as ButtonGroup with a single button is not rendered")
         
         ///This below doesn't work