Correctly handling zk client loss