Fix a layout issue in the generator