Made the test example a bit more robust.