small test refactor