From 63e1153d4a0cdbf92302a6349cd1ae23104b3f6e Mon Sep 17 00:00:00 2001
From: Marten Seemann <martenseemann@gmail.com>
Date: Mon, 8 Jun 2020 08:52:30 +0700
Subject: [PATCH] fix debug message when returning an early session

---
 client.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/client.go b/client.go
index 66a9b713..322918e3 100644
--- a/client.go
+++ b/client.go
@@ -81,8 +81,12 @@ func DialAddrEarly(
 	tlsConf *tls.Config,
 	config *Config,
 ) (EarlySession, error) {
-	defer utils.Logger.WithPrefix(utils.DefaultLogger, "client").Debugf("Returning early session")
-	return dialAddrContext(context.Background(), addr, tlsConf, config, true)
+	sess, err := dialAddrContext(context.Background(), addr, tlsConf, config, true)
+	if err != nil {
+		return nil, err
+	}
+	utils.Logger.WithPrefix(utils.DefaultLogger, "client").Debugf("Returning early session")
+	return sess, nil
 }
 
 // DialAddrContext establishes a new QUIC connection to a server using the provided context.