Made the test a little more robust.