tests: minor test refactor