fix zk bug