Minor controller fix