diff --git a/mace/core/registry.h b/mace/core/registry.h
index 277cabb3ec3f3c854d80d4c095643d9f59f547e5..ac81328731c3178ee19798cae862086f25ed8c29 100644
--- a/mace/core/registry.h
+++ b/mace/core/registry.h
@@ -15,6 +15,7 @@
 #ifndef MACE_CORE_REGISTRY_H_
 #define MACE_CORE_REGISTRY_H_
 
+#include <functional>
 #include <map>
 #include <memory>
 #include <mutex>  // NOLINT(build/c++11)
diff --git a/mace/kernels/proposal.h b/mace/kernels/proposal.h
index 748264e5fe503884804da023d4064e3d82a98d5c..89f79b7fa702a05de8b7d781c380420915a2ca20 100644
--- a/mace/kernels/proposal.h
+++ b/mace/kernels/proposal.h
@@ -16,6 +16,7 @@
 #define MACE_KERNELS_PROPOSAL_H_
 
 #include <algorithm>
+#include <cmath>
 #include <vector>
 
 #include "mace/core/future.h"