Minor controller update