small change to controller