Add missing notification registration