Fix notification tests