1 files changed, 4 insertions, 2 deletions
diff --git a/internal/ssh/client/customkeycallback.go b/internal/ssh/client/customkeycallback.go
index 53b8e3c..0107895 100644
--- a/internal/ssh/client/customkeycallback.go
+++ b/internal/ssh/client/customkeycallback.go
@@ -1,6 +1,7 @@
 package client
 
 import (
+	"context"
 	"net"
 
 	"golang.org/x/crypto/ssh"
@@ -15,8 +16,9 @@ func NewCustomCallback() (*CustomCallback, error) {
 	return &h, nil
 }
 
-// Wrap the host key callback.
-func (h *CustomCallback) Wrap() ssh.HostKeyCallback {
+// Wrap the host key callback. ctx is accepted for interface compatibility
+// but the custom callback never blocks so it has nothing to abort.
+func (h *CustomCallback) Wrap(_ context.Context) ssh.HostKeyCallback {
 	return func(server string, remote net.Addr, key ssh.PublicKey) error {
 		return nil
 	}