Fix to mallet generator.