diff --git a/include/control_toolbox/pid.hpp b/include/control_toolbox/pid.hpp
index 8fe251c..a71bae3 100644
--- a/include/control_toolbox/pid.hpp
+++ b/include/control_toolbox/pid.hpp
@@ -33,10 +33,7 @@
 #ifndef CONTROL_TOOLBOX__PID_HPP_
 #define CONTROL_TOOLBOX__PID_HPP_
 
-#include <iostream>
-#include <memory>
-#include <stdexcept>
-#include <string>
+#include <chrono>
 
 #include "rclcpp/duration.hpp"
 #include "realtime_tools/realtime_buffer.hpp"
diff --git a/src/pid.cpp b/src/pid.cpp
index 8aed6a3..26b0915 100644
--- a/src/pid.cpp
+++ b/src/pid.cpp
@@ -38,9 +38,9 @@
 
 #include <algorithm>
 #include <cmath>
-#include <string>
+#include <iostream>
+#include <stdexcept>
 #include <utility>
-#include <vector>
 
 #include "control_toolbox/pid.hpp"