fix a bug of ZhiHeng AI