Fix data_generator params