Minor update to ensure correct generation of Java