The fix improves the test robustness by making it retry once in case of IO/Connect exception. Reviewed-by: chegar