diff --git a/lib/generate.sh b/lib/generate.sh
index 7029123c..79b09414 100755
--- a/lib/generate.sh
+++ b/lib/generate.sh
@@ -8,14 +8,6 @@ fi
 
 # start the YANG array
 MAIN_YANG_ARRAY="
-struct np2_file {
-    const char *file;
-    const char *name;
-    const char *revision;
-    const char *data;
-    int len;
-};
-
 struct np2_file yang_files[] = {
 "
 
diff --git a/lib/np2_sr_setup.c b/lib/np2_sr_setup.c
index 1db5d32f..6a2bc70a 100644
--- a/lib/np2_sr_setup.c
+++ b/lib/np2_sr_setup.c
@@ -16,6 +16,8 @@
 
 #define _GNU_SOURCE
 
+#include "netopeer2.h"
+
 #include <stdint.h>
 #include <stdio.h>
 #include <stdlib.h>