task 4 - fix