Fix the Edit button by adding the controller to the button.