diff --git a/pyproject.toml b/pyproject.toml
index 15b0a17e2..2680f3464 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -35,20 +35,15 @@ dependencies = [
   "sympy>=1.12",
 ]
 
-# TODO: update relay-bp to
-# "relay-bp[stim]==0.2.?",
-# once https://github.com/trmue/relay/pull/30 is in a release
 [project.optional-dependencies]
 dev = [
   "checks-superstaq>=0.5.45",
   "jupyter>=1.1.1",
   "python-lsp-server[all]>=1.14.0",
-  # "relay-bp[stim]==0.1.0",
-  "relay-bp[stim] @ git+https://github.com/perlinm/relay-bp.git",
+  "relay-bp[stim]==0.2.1",
 ]
 relay-bp = [
-  # "relay-bp[stim]==0.1.0",
-  "relay-bp[stim] @ git+https://github.com/perlinm/relay-bp.git",
+  "relay-bp[stim]==0.2.1",
 ]
 
 [project.urls]
diff --git a/src/qldpc/decoders/custom.py b/src/qldpc/decoders/custom.py
index 1ad400807..aaaf065a4 100644
--- a/src/qldpc/decoders/custom.py
+++ b/src/qldpc/decoders/custom.py
@@ -61,7 +61,7 @@ class RelayBPDecoder(BatchDecoder):
 
     This class first constructs a relay_bp.decoder.DynDecoder decoder by class name, such as
     "RelayDecoderF32"; see help(relay_bp) for more options.  To enable parallelized decoding, which
-    which as of relay-bp==0.1.0 is only implemented for the relay_bp.ObservableDecoderRunner class,
+    which as of relay-bp==0.2.1 is only implemented for the relay_bp.ObservableDecoderRunner class,
     RelayBPDecoder wraps the relay_bp.decoder.DynDecoder in a relay_bp.ObservableDecoderRunner at
     initialization time.