diff --git a/src/eapol.c b/src/eapol.c
index da4622fc..bd5155d9 100644
--- a/src/eapol.c
+++ b/src/eapol.c
@@ -451,3 +451,8 @@ 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));
 }
+
+void eapol_sm_set_aa_address(struct eapol_sm *sm, const uint8_t *aa_addr)
+{
+	memcpy(sm->aa_addr, aa_addr, sizeof(sm->aa_addr));
+}
diff --git a/src/eapol.h b/src/eapol.h
index 1aa0bd09..f499bac2 100644
--- a/src/eapol.h
+++ b/src/eapol.h
@@ -126,3 +126,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);
+void eapol_sm_set_aa_address(struct eapol_sm *sm, const uint8_t *aa_addr);