diff --git a/src/eapol.c b/src/eapol.c
index b9f4732f..2e0e80cd 100644
--- a/src/eapol.c
+++ b/src/eapol.c
@@ -493,6 +493,11 @@ void __eapol_set_get_nonce_func(eapol_get_nonce_func_t func)
 	get_nonce = func;
 }
 
+void __eapol_set_protocol_version(enum eapol_protocol_version version)
+{
+	protocol_version = version;
+}
+
 bool eapol_init()
 {
 	state_machines = l_hashmap_new();
diff --git a/src/eapol.h b/src/eapol.h
index a7cbf3c0..606488ee 100644
--- a/src/eapol.h
+++ b/src/eapol.h
@@ -129,6 +129,7 @@ struct eapol_key *eapol_create_ptk_4_of_4(
 
 void __eapol_set_tx_packet_func(eapol_tx_packet_func_t func);
 void __eapol_set_get_nonce_func(eapol_get_nonce_func_t func);
+void __eapol_set_protocol_version(enum eapol_protocol_version version);
 
 struct eapol_sm *eapol_sm_new();
 void eapol_sm_free(struct eapol_sm *sm);