Fix a test that failed sometimes