Fix notification tests again