Remove junit dependency in generated code