Fix the notification tests