diff --git a/selfdrive/controls/lib/disturbance_controller.py b/selfdrive/controls/lib/disturbance_controller.py
index d3b66673b..2567a5fd9 100644
--- a/selfdrive/controls/lib/disturbance_controller.py
+++ b/selfdrive/controls/lib/disturbance_controller.py
@@ -1,6 +1,7 @@
 import numpy as np
 import math
 
+from openpilot.common.pid import PIDController
 from openpilot.common.realtime import DT_CTRL
 
 ALPHA_MIN = 0.004