diff --git a/src/binding.cc b/src/binding.cc
index 668ab933d67a047e425d9268d101d713eea2bcea..445f853f328cf6183e98cc2f86c631ba1056890c 100644
--- a/src/binding.cc
+++ b/src/binding.cc
@@ -1,4 +1,4 @@
-#include <bits/stdc++.h>
+#include <vector>
 #include <pybind11/pybind11.h>
 #include <pybind11/pytypes.h>
 #include <pybind11/numpy.h>
diff --git a/src/nms.cc b/src/nms.cc
index d83fae454e65a7854b81cc2079dd05974272cdc9..84122685c129483d605d33344b49cf5cb27672c2 100644
--- a/src/nms.cc
+++ b/src/nms.cc
@@ -1,4 +1,5 @@
-#include <bits/stdc++.h>
+#include <cmath>
+#include <algorithm>
 
 namespace nms
 {