[TASK] Fix failing test