cleaned a hack in datagen