Fix for ControllerGenTest.