Fix notification regression