Correcting issue with notification