diff --git a/AugFuncs.h b/AugFuncs.h
index b923732..fff082c 100644
--- a/AugFuncs.h
+++ b/AugFuncs.h
@@ -1,5 +1,6 @@
 #include "AugSettings.h"
 #include "augeas.h"
+#include "map"
 
 #ifndef BLUEWEST_AUGEAS_BLUEAUG_H
 #define BLUEWEST_AUGEAS_BLUEAUG_H
diff --git a/main.cpp b/main.cpp
index 7cfcd37..067660d 100644
--- a/main.cpp
+++ b/main.cpp
@@ -1,8 +1,4 @@
 #include <iostream>
-#include <exception>
-#include <augeas.h>
-#include <vector>
-#include <map>
 #include "AugSettings.h"
 #include "AugFuncs.h"