minor change to controller