diff --git a/docs/overrides/main.html b/docs/overrides/main.html
index 9114c614..01fd277c 100644
--- a/docs/overrides/main.html
+++ b/docs/overrides/main.html
@@ -1,12 +1,12 @@
 {% extends "base.html" %}
-
 {% block content %}
 {% if page.nb_url %}
-    <a href="{{ page.nb_url }}" title="Download Notebook" class="md-content__button md-icon">
+    {% set path_parts = page.url.strip('/').split('/') %}
+    {% set last_part = path_parts[-1] %}
+    {% set notebook_url = page.url ~ last_part ~ '.ipynb' %}
+    <a href="{{ notebook_url | url }}" title="Download Notebook" class="md-content__button md-icon">
         {% include ".icons/material/download.svg" %}
     </a>
-
 {% endif %}
-
 {{ super() }}
-{% endblock content %}
+{% endblock content %}
\ No newline at end of file