修正一个和数据集相关的bug