Update TestGenEntity.java