Little fix in Task 004/01