Fixed bug in datagen.