diff --git a/server/server_test.go b/server/server_test.go
index 5e10191b51d99ae197264b2c43e757ce83238531..021ed810164912711721bd727d7ef867ac9a0161 100644
--- a/server/server_test.go
+++ b/server/server_test.go
@@ -44,7 +44,7 @@ func TestServer_OpenAndGet(t *testing.T) {
 	kc := c.client(t)
 
 	// Decrypt the private key with the right password.
-	err := kc.Open(context.Background(), "", "testuser", string(pw), 60)
+	err := kc.Open(context.Background(), "", "testuser", string(pw), "session", 60)
 	if err != nil {
 		t.Fatal("keystore.Open():", err)
 	}
@@ -63,7 +63,7 @@ func TestServer_OpenAndGet(t *testing.T) {
 	}
 
 	// Call Close and forget the key.
-	if err := kc.Close(context.Background(), "", "testuser"); err != nil {
+	if err := kc.Close(context.Background(), "", "testuser", "session"); err != nil {
 		t.Fatalf("keystore.Close(): %v", err)
 	}
 	if _, err := kc.Get(context.Background(), "", "testuser", ssoTicket); err != client.ErrNoKeys {
@@ -78,7 +78,7 @@ func TestServer_OpenAndGet_NoKeys(t *testing.T) {
 
 	// Check the return value of Open() when the user has no keys.
 	username := "no-keys-user"
-	err := kc.Open(context.Background(), "", username, string(pw), 60)
+	err := kc.Open(context.Background(), "", username, string(pw), "session", 60)
 	if err != nil {
 		t.Fatalf("keystore.Open(%s): unexpected err=%v", username, err)
 	}