diff --git a/.sage/go.mod b/.sage/go.mod
index 17f02c9..bec1147 100644
--- a/.sage/go.mod
+++ b/.sage/go.mod
@@ -2,4 +2,4 @@ module sage
 
 go 1.17
 
-require go.einride.tech/sage v0.328.0
+require go.einride.tech/sage v0.331.0
diff --git a/.sage/go.sum b/.sage/go.sum
index 1522acd..b719cbe 100644
--- a/.sage/go.sum
+++ b/.sage/go.sum
@@ -1,2 +1,2 @@
-go.einride.tech/sage v0.328.0 h1:72XhWkfj8HWIQGeZVu5IfRiBngEycn+D8lCGSl5aAk0=
-go.einride.tech/sage v0.328.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=