diff --git a/augmentation.h b/augmentation.h
index 48ec595f..63db47a3 100644
--- a/augmentation.h
+++ b/augmentation.h
@@ -11,7 +11,7 @@
 #ifndef augmentation_h
 #define augmentation_h
 
-#include <linux/inet.h>
+#include <linux/kref.h>
 
 typedef struct augmentation_response
 {