diff --git a/io/iom_wait.c b/io/iom_wait.c
index d58afa9..ff59990 100644
--- a/io/iom_wait.c
+++ b/io/iom_wait.c
@@ -8,6 +8,11 @@
 #include <sys/time.h>
 #endif
 #include <errno.h>
+#ifdef __dietlibc__
+#include <threads.h>
+#else
+#include <semaphore.h>
+#endif
 
 // return the next event, waiting of none are queued
 // wait at most timeout milliseconds