diff --git a/kernel/ethosu_mailbox.h b/kernel/ethosu_mailbox.h
index 5a77b54..8f539ee 100644
--- a/kernel/ethosu_mailbox.h
+++ b/kernel/ethosu_mailbox.h
@@ -104,6 +104,9 @@
 			     struct ethosu_buffer **ifm,
 			     uint32_t ofm_count,
 			     struct ethosu_buffer **ofm,
-			     struct ethosu_buffer *network);
+			     struct ethosu_buffer *network,
+			     uint8_t *pmu_event_config,
+			     uint8_t pmu_event_config_count,
+			     uint8_t pmu_cycle_counter_enable);
 
 #endif /* ETHOSU_MAILBOX_H */