[TASK] Fix a failing test