fix the controller setup