diff --git a/kernel/ethosu_inference.c b/kernel/ethosu_inference.c
index bc38236..4dcdc0a 100644
--- a/kernel/ethosu_inference.c
+++ b/kernel/ethosu_inference.c
@@ -166,6 +166,7 @@
 		ethosu_buffer_put(inf->ofm[inf->ofm_count]);
 
 	ethosu_network_put(inf->net);
+	memset(inf, 0, sizeof(*inf));
 	devm_kfree(dev, inf);
 }
 
@@ -384,6 +385,7 @@
 		ethosu_buffer_put(inf->ifm[inf->ifm_count]);
 
 kfree:
+	memset(inf, 0, sizeof(*inf));
 	devm_kfree(dev, inf);
 
 	return ret;