Made the test slightly more robust