diff --git a/session.go b/session.go
index fbe5692..3a3ad70 100644
--- a/session.go
+++ b/session.go
@@ -2,7 +2,6 @@ package ssh
 
 import (
 	"bytes"
-	"context"
 	"errors"
 	"fmt"
 	"net"
@@ -60,7 +59,7 @@ type Session interface {
 	//
 	// The context is canceled when the client's connection closes or I/O
 	// operation fails.
-	Context() context.Context
+	Context() Context
 
 	// Permissions returns a copy of the Permissions object that was available for
 	// setup in the auth handlers via the Context.
@@ -159,7 +158,7 @@ func (sess *session) Permissions() Permissions {
 	return *perms
 }
 
-func (sess *session) Context() context.Context {
+func (sess *session) Context() Context {
 	return sess.ctx
 }