another little change on the controller