diff --git a/kernel/ethosu_mailbox.h b/kernel/ethosu_mailbox.h
index ea4409f..a3e2c14 100644
--- a/kernel/ethosu_mailbox.h
+++ b/kernel/ethosu_mailbox.h
@@ -140,11 +140,12 @@
 int ethosu_mailbox_pong(struct ethosu_mailbox *mbox);
 
 /**
- * ethosu_mailbox_version_response - Send version request
+ * ethosu_mailbox_version_request() - Send protocol version request
  *
  * Return: 0 on succes, else error code
  */
-int ethosu_mailbox_version_request(struct ethosu_mailbox *mbox);
+int ethosu_mailbox_version_request(struct ethosu_mailbox *mbox,
+				   struct ethosu_mailbox_msg *msg);
 
 /**
  * ethosu_mailbox_capabilities_request() - Send capabilities request