Update to show notification correctly