Update MockGenerator.java