diff --git a/driver/LKM/src/smith_hook.c b/driver/LKM/src/smith_hook.c
index e6656f8b8..252290cb2 100644
--- a/driver/LKM/src/smith_hook.c
+++ b/driver/LKM/src/smith_hook.c
@@ -14,7 +14,12 @@
 #include <linux/moduleparam.h>
 #include <linux/netfilter.h>
 #include <linux/netfilter_ipv4.h>
+
+#if IS_ENABLED(CONFIG_IPV6)
+#include <linux/ipv6.h>
+#include <net/ipv6.h> /* ipv6_addr_any */
 #include <linux/netfilter_ipv6.h>
+#endif
 
 #define EXIT_PROTECT 0
 #define SANDBOX 0
diff --git a/driver/LKM/src/trace_buffer.c b/driver/LKM/src/trace_buffer.c
index dfe93a7be..ff4148b64 100644
--- a/driver/LKM/src/trace_buffer.c
+++ b/driver/LKM/src/trace_buffer.c
@@ -26,6 +26,7 @@
 #include <linux/oom.h>
 #include <linux/kernel.h>
 
+#include "../include/util.h"
 #include "../include/trace_buffer.h"
 
 /* kernel has READ_ONCE defined since 3.18.13 */