Fix notify callback problem