Update controller: minor change