Fix notification bug