[TASK] Updated tests