Modify how to generate test data