test: refactor test