Revert the controller code