Fix test that seems to fail every once in a while