Fixes notifications state restoration

Fixes #2439

/cc @sendhil