Fixed controller class generation