Update to fix notify