Fix failing notification tests