diff --git a/docs/conf.py b/docs/conf.py
index d1902ab308c..0e01b3164e4 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -114,10 +114,10 @@
 # TODO: branch/tag should change depending on which version of docs you look at
 # TODO: width option of image directive is broken, see:
 # https://github.com/pytorch/pytorch_sphinx_theme/issues/140
-nbsphinx_prolog = f"""
+nbsphinx_prolog = """
 {% set colab = "https://colab.research.google.com" %}
 {% set repo = "microsoft/torchgeo" %}
-{% set branch = "releases/v{version}" %}
+{% set branch = "releases/v{{ env.config.version }}" %}
 
 .. image:: {{ colab }}/assets/colab-badge.svg
    :class: colabbadge