1 files changed, 4 insertions, 2 deletions
diff --git a/corking.h b/corking.h
index b240595..fb4483c 100644
--- a/corking.h
+++ b/corking.h
@@ -1,11 +1,13 @@
 #ifndef CORKING_H
 #define CORKING_H
 
+#include <stdbool.h>
+
 extern void set_tcp_cork(int fd);
 extern void set_tcp_uncork(int fd);
 extern void set_udp_cork(int fd);
 extern void set_udp_uncork(int fd);
-extern void set_sock_cork(int fd, int udp);
-extern void set_sock_uncork(int fd, int udp);
+extern void set_sock_cork(int fd, bool is_udp);
+extern void set_sock_uncork(int fd, bool is_udp);
 
 #endif /* CORKING_H */