--- a/mem2swap.c
+++ b/mem2swap.c
@@ -9,7 +9,7 @@
 #include <stdlib.h>
 #include <unistd.h>
 #include <errno.h>
-#include <wait.h>
+#include <sys/wait.h>
 
 #define N 1000