diff --git a/kernel/ethosu_inference.c b/kernel/ethosu_inference.c
index 50007d0..6d3405d 100644
--- a/kernel/ethosu_inference.c
+++ b/kernel/ethosu_inference.c
@@ -385,6 +385,8 @@
 	while (inf->ifm_count-- > 0)
 		ethosu_buffer_put(inf->ifm[inf->ifm_count]);
 
+	ethosu_mailbox_deregister(mailbox, &inf->msg);
+
 kfree:
 	memset(inf, 0, sizeof(*inf));
 	devm_kfree(dev, inf);