More idiomatic testing