refactor: split tests into two tests