Make a small change to generation code