diff --git a/penzai/__init__.py b/penzai/__init__.py
index 03bc5c8..d56dfd7 100644
--- a/penzai/__init__.py
+++ b/penzai/__init__.py
@@ -14,4 +14,4 @@
 
 """A JAX research toolkit for building, editing, and visualizing neural networks."""
 
-__version__ = '0.2.1'
+__version__ = '0.2.2'
diff --git a/pyproject.toml b/pyproject.toml
index 2fac10e..e41e1a3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,7 +23,7 @@ dependencies = [
     "jax>=0.4.23",
     "numpy>=1.25.2",
     "ordered_set>=4.1.0",
-    "treescope>=0.1.0",
+    "treescope>=0.1.3",
     "typing_extensions>=4.2",
 ]