Extend NotificationPlugin instead