Reverting accidentally removed notification code