[TASK] Add test