Update ControllerBase.php