more idiomatic unit test