Move the notification callback to an action