diff --git a/lib/stmp_device.c b/lib/stmp_device.c
index 8ac9bcc4289a65326437d632d21b1f4904878c98..a904656f4fd7c5848155feaf0b0b3b4b50eeb00c 100644
--- a/lib/stmp_device.c
+++ b/lib/stmp_device.c
@@ -15,7 +15,8 @@
 #include <linux/io.h>
 #include <linux/errno.h>
 #include <linux/delay.h>
-#include <linux/module.h>
+#include <linux/compiler.h>
+#include <linux/export.h>
 #include <linux/stmp_device.h>
 
 #define STMP_MODULE_CLKGATE	(1 << 30)