Minor fix to iteration 1