diff --git a/pyproject.toml b/pyproject.toml
index b030772..dc2a2f2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -33,5 +33,5 @@ dependencies= [
     "torch>=2.1.0",
 
     # Agogos
-    "agogos==0.3.4",
+    "agogos==0.4",
 ]
diff --git a/requirements.txt b/requirements.txt
index 38c6e35..88ec062 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-agogos==0.3.4
+agogos==0.4
 annotated-types==0.6.0
 certifi==2024.2.2
 charset-normalizer==3.3.2