mk5 - Minor fix

task 7 HW 2