Update TestGenerator.java