TASK-13 Fixing broken test.