diff --git a/TAP.h b/TAP.h
index b095249..eeb6a14 100644
--- a/TAP.h
+++ b/TAP.h
@@ -5,10 +5,9 @@
 #include <string.h>
 #include <unistd.h>
 #include <sys/ioctl.h>
-#include <linux/if.h>
+#include <net/if.h>
 #include <linux/if_tun.h>
-//#include <sys/socket.h>
-//#include <netinet/in.h>
+#include <sys/socket.h>
 #include <arpa/inet.h>
 #include "Constants.h"
 
diff --git a/makefile b/makefile
index 5ff4f30..514e245 100644
--- a/makefile
+++ b/makefile
@@ -2,7 +2,7 @@
 .PHONY: all clean install tncattach
 
 compiler = gcc
-flags = -lm
+flags = -std=gnu11 -lm
 
 all: tncattach
 rebuild: clean all