Java codegen fixes