diff --git a/src/include/gotty/tty.h b/src/include/gotty/tty.h
index 136bdc5..ad09d4d 100644
--- a/src/include/gotty/tty.h
+++ b/src/include/gotty/tty.h
@@ -28,5 +28,6 @@
 void configure_stdout(void);
 void restore_stdout(void);
 int connect_tty(void);
+void wait_for_tty_device(void);
 
 #endif
diff --git a/src/main.c b/src/main.c
index 1f72925..27a06a3 100644
--- a/src/main.c
+++ b/src/main.c
@@ -20,6 +20,7 @@
  */
 
 #include <stdio.h>
+#include <stdlib.h>
 #include "gotty/options.h"
 #include "gotty/tty.h"