diff --git a/auth_test.go b/auth_test.go
index 4692557..485af4d 100644
--- a/auth_test.go
+++ b/auth_test.go
@@ -28,8 +28,8 @@ func TestAuthWhitelist(t *testing.T) {
 	}
 
 	auth := NewAuth()
-	ok, err := auth.Check(nil, key)
-	if !ok || err != nil {
+	err = auth.Check(nil, key, "")
+	if err != nil {
 		t.Error("Failed to permit in default state:", err)
 	}
 
@@ -44,8 +44,8 @@ func TestAuthWhitelist(t *testing.T) {
 		t.Error("Clone key does not match.")
 	}
 
-	ok, err = auth.Check(nil, keyClone)
-	if !ok || err != nil {
+	err = auth.Check(nil, keyClone, "")
+	if err != nil {
 		t.Error("Failed to permit whitelisted:", err)
 	}
 
@@ -54,9 +54,8 @@ func TestAuthWhitelist(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	ok, err = auth.Check(nil, key2)
-	if ok || err == nil {
+	err = auth.Check(nil, key2, "")
+	if err == nil {
 		t.Error("Failed to restrict not whitelisted:", err)
 	}
-
 }
diff --git a/sshd/client_test.go b/sshd/client_test.go
index 80499ca..1c0ead4 100644
--- a/sshd/client_test.go
+++ b/sshd/client_test.go
@@ -15,8 +15,8 @@ type RejectAuth struct{}
 func (a RejectAuth) AllowAnonymous() bool {
 	return false
 }
-func (a RejectAuth) Check(net.Addr, ssh.PublicKey) (bool, error) {
-	return false, errRejectAuth
+func (a RejectAuth) Check(net.Addr, ssh.PublicKey, string) error {
+	return errRejectAuth
 }
 
 func TestClientReject(t *testing.T) {