diff --git a/client.go b/client.go
index 322918e3..5d542c57 100644
--- a/client.go
+++ b/client.go
@@ -144,7 +144,7 @@ func DialEarly(
 	host string,
 	tlsConf *tls.Config,
 	config *Config,
-) (Session, error) {
+) (EarlySession, error) {
 	return dialContext(context.Background(), pconn, remoteAddr, host, tlsConf, config, true, false)
 }