diff --git a/test_library.c b/test_library.c
index 7bb901b..55d3f49 100644
--- a/test_library.c
+++ b/test_library.c
@@ -15,6 +15,9 @@
 
 #define _GNU_SOURCE
 
+#include <unistd.h>
+
+#include "util.h"
 #include "wattsup_common.h"
 #include "globals.h"
 
diff --git a/util.h b/util.h
index 70a5ce8..07dd835 100644
--- a/util.h
+++ b/util.h
@@ -1,4 +1,16 @@
 #ifndef UTIL_H
 #define UTIL_H
 
+int parse_args(int argc, char ** argv);
+int wu_check_store(int index, int dev_fd);
+int open_device(char * device_name, int * dev_fd);
+void dbg(const char * fmt, ...);
+int setup_serial_device(int dev_fd);
+int wu_clear(int fd);
+static int wu_show_cal(int fd);
+int wu_check_show(int index, int dev_fd);
+int wu_start_log(void);
+int wu_read_data(int fd);
+int wu_stop_log(void);
+int wu_show(int index, int dev_fd);
 #endif
diff --git a/wattsup.c b/wattsup.c
index 966af67..d943869 100644
--- a/wattsup.c
+++ b/wattsup.c
@@ -10,7 +10,10 @@
 
 #define _GNU_SOURCE
 
+#include <unistd.h>
+
 #include "wattsup_common.h"
+#include "util.h"
 #include "globals.h"
 
 int main(int argc, char ** argv)