Minor fix in initial function