diff --git a/unit/test-eapol.c b/unit/test-eapol.c
index e1b6c96e..86b154eb 100644
--- a/unit/test-eapol.c
+++ b/unit/test-eapol.c
@@ -2471,7 +2471,7 @@ static void eapol_sm_test_wpa_ptk_gtk_2(const void *data)
 	eapol_exit();
 }
 
-static void verify_install_tk(struct handshake_state *hs,
+static void verify_install_tk(struct handshake_state *hs, uint8_t key_index,
 				const uint8_t *tk, uint32_t cipher)
 {
 	struct test_handshake_state *ths =
@@ -3626,7 +3626,7 @@ static void test_ap_sta_hs_event(struct handshake_state *hs,
 	assert(event != HANDSHAKE_EVENT_FAILED);
 }
 
-static void test_ap_sta_install_tk(struct handshake_state *hs,
+static void test_ap_sta_install_tk(struct handshake_state *hs, uint8_t key_idx,
 					const uint8_t *tk, uint32_t cipher)
 {
 	struct test_ap_sta_hs *ths =
diff --git a/unit/test-wsc.c b/unit/test-wsc.c
index 7b24e6c5..8b130f45 100644
--- a/unit/test-wsc.c
+++ b/unit/test-wsc.c
@@ -2345,7 +2345,7 @@ static bool random_nonce(uint8_t nonce[])
 	return l_getrandom(nonce, 32);
 }
 
-static void test_ap_sta_install_tk(struct handshake_state *hs,
+static void test_ap_sta_install_tk(struct handshake_state *hs, uint8_t key_idx,
 					const uint8_t *tk, uint32_t cipher)
 {
 	assert(false);