slightly better test structure, but possibly not idiomatic