diff --git a/pyproject.toml b/pyproject.toml
index 4600995..15f1bd5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sphinxcontrib-d2lang"
-version = "0.0.3"
+version = "0.0.4"
 authors = [
   { name="Milka64", email="michael.ricart@0w.tf" },
 ]
diff --git a/setup.py b/setup.py
index 05bdb76..d689b35 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@ with open('README.md', 'r', encoding='utf8') as file:
 
 setup(
     name='sphinxcontrib-d2lang',
-    version='0.0.3',
+    version='0.0.4',
     author='Milka64',
     author_email='michael.ricart@0w.tf',
     url='https://git.0w.tf/Milka64/sphinx-d2lang/',
diff --git a/sphinxcontrib/d2lang/d2lang.py b/sphinxcontrib/d2lang/d2lang.py
index 28914bf..20fb651 100644
--- a/sphinxcontrib/d2lang/d2lang.py
+++ b/sphinxcontrib/d2lang/d2lang.py
@@ -22,10 +22,11 @@ class D2langDirective(SphinxDirective):
         d2_bin = which('d2')
         srcdir = self.state.document.settings.env.srcdir
         diag_source = self.content
+        out_dir = self.state.document.current_source.rsplit("/",1)[0]
         if 'filename' in self.options:
-            output_fname = self.options.get('filename')
+            output_fname = out_dir + "/" + self.options.get('filename')
         else:
-            output_fname = "d2lang_svg/" + str(uuid.uuid4()) + ".svg"
+            output_fname = out_dir + "/" + "d2lang_svg/" + str(uuid.uuid4()) + ".svg"
         if 'layout' in self.options:
             layout = self.options.get('layout')
         else: