[TASK] Fix broken test