[TASK] Cover newly added code with tests