[TASK] Fix category collection functional test