Minor refactor to make test more explicit