Minor fix to Controller