diff --git a/docs/source/conf.py b/docs/source/conf.py
index bc6d2da..e2a7d73 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -59,6 +59,9 @@
 autodoc_typehints = "description"
 autodoc_typehints_description_target = "documented"
 simplify_optional_unions = False
+autodoc_typehints_format = 'short'
+python_use_unqualified_type_names = True
+
 
 # -- Options for viewCode extension -------------------------------------------
 viewcode_line_numbers = True
@@ -165,7 +168,7 @@
 versions.append(("latest", "https://www.sachahu.com/docs/diffusion-model-framework/latest"))
 
 html_context = {
-  "current_version": "0.0.1",
+  "current_version": release,
   "versions": versions,
 }