Some modifications to sbt/test-generate