Small refactor of test.