fix test data generator