diff --git a/crypto.cc b/crypto.cc
index 18088eb..b3ada92 100644
--- a/crypto.cc
+++ b/crypto.cc
@@ -1,5 +1,6 @@
 #include <arpa/inet.h>
 #include <netinet/in.h>
+#include <signal.h>
 #include <stdio.h>
 #include <unistd.h>
 
diff --git a/tlv.cc b/tlv.cc
index 58a56c3..dbb8774 100644
--- a/tlv.cc
+++ b/tlv.cc
@@ -1,3 +1,5 @@
+#include <arpa/inet.h>
+
 #include <cassert>
 #include <iostream>
 
diff --git a/tlv.h b/tlv.h
index f46894a..0432cfe 100644
--- a/tlv.h
+++ b/tlv.h
@@ -1,6 +1,7 @@
 #include <stdint.h>
 
 #include <list>
+#include <memory>
 #include <string>
 
 class TLVNode {