Much nicer, and actually functional, unit test.