Fix test case for notification_service