Modify zk notify logic