Update the notification tests to work with Jinja