Minor fix in setup