Fix to work with the updated PID controller