diff --git a/.sage/go.mod b/.sage/go.mod
index 6f40e27..786d5fb 100644
--- a/.sage/go.mod
+++ b/.sage/go.mod
@@ -2,4 +2,4 @@ module go.einride.tech/pid/.sage
 
 go 1.19
 
-require go.einride.tech/sage v0.330.0
+require go.einride.tech/sage v0.331.0
diff --git a/.sage/go.sum b/.sage/go.sum
index de2a491..b719cbe 100644
--- a/.sage/go.sum
+++ b/.sage/go.sum
@@ -1,2 +1,2 @@
-go.einride.tech/sage v0.330.0 h1:N7xpuNUkwBQpwSiPwlZEgqV8E9SUah22XFe3ygCM39Q=
-go.einride.tech/sage v0.330.0/go.mod h1:EzV5uciFX7/2ho8EKB5K9JghOfXIxlzs694b+Tkl5GQ=
+go.einride.tech/sage v0.331.0 h1:G06O2Wm/CRq6czAfhq8ctMucGIqrbeor7Vq8CWVWRQY=
+go.einride.tech/sage v0.331.0/go.mod h1:EzV5uciFX7/2ho8EKB5K9JghOfXIxlzs694b+Tkl5GQ=