diff --git a/src/eapol.c b/src/eapol.c
index f6cee970..da4622fc 100644
--- a/src/eapol.c
+++ b/src/eapol.c
@@ -446,3 +446,8 @@ void eapol_sm_free(struct eapol_sm *sm)
 {
 	eapol_sm_destroy(sm);
 }
+
+void eapol_sm_set_sta_address(struct eapol_sm *sm, const uint8_t *sta_addr)
+{
+	memcpy(sm->sta_addr, sta_addr, sizeof(sm->sta_addr));
+}
diff --git a/src/eapol.h b/src/eapol.h
index f94beb1d..1aa0bd09 100644
--- a/src/eapol.h
+++ b/src/eapol.h
@@ -124,3 +124,5 @@ struct eapol_key *eapol_create_ptk_4_of_4(
 
 struct eapol_sm *eapol_sm_new();
 void eapol_sm_free(struct eapol_sm *sm);
+
+void eapol_sm_set_sta_address(struct eapol_sm *sm, const uint8_t *sta_addr);