Properly wrap TimeoutError with MaxRetryError.