[TASK] Fix functional tests