diff --git a/drivers/misc/sram-exec.c b/drivers/misc/sram-exec.c
index ac522417c46296685b5287ed2bddb3bd036ecf11..3d528a13b8fc657d0400aec40a0de36a424aa9db 100644
--- a/drivers/misc/sram-exec.c
+++ b/drivers/misc/sram-exec.c
@@ -16,9 +16,10 @@
 
 #include <linux/device.h>
 #include <linux/genalloc.h>
+#include <linux/mm.h>
 #include <linux/sram.h>
 
-#include <asm/cacheflush.h>
+#include <asm/set_memory.h>
 
 #include "sram.h"