diff --git a/arch/microblaze/kernel/heartbeat.c b/arch/microblaze/kernel/heartbeat.c
index 2022130139d2..87aa942eee8c 100644
--- a/arch/microblaze/kernel/heartbeat.c
+++ b/arch/microblaze/kernel/heartbeat.c
@@ -11,10 +11,10 @@
 #include <linux/sched.h>
 #include <linux/sched/loadavg.h>
 #include <linux/io.h>
+#include <linux/of.h>
 
 #include <asm/setup.h>
 #include <asm/page.h>
-#include <asm/prom.h>
 
 static unsigned int base_addr;