fix the notification bug