Minor fix on controller generator