Making the test a little less brittle