Fix test generator for controller