Minor fix in setup import