diff --git a/pyproject.toml b/pyproject.toml
index b622aab..c490885 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -27,7 +27,7 @@ keywords = [
 name = "FlowerPower"
 readme = "README.md"
 requires-python = ">= 3.11"
-version = "0.8.5"
+version = "0.8.5.1"
 
 [project.scripts]
 flowerpower = "flowerpower.cli:app"
diff --git a/src/flowerpower/pipeline.py b/src/flowerpower/pipeline.py
index 531f759..e0cc6be 100644
--- a/src/flowerpower/pipeline.py
+++ b/src/flowerpower/pipeline.py
@@ -1119,7 +1119,9 @@ def save_dag(
         dag = self._display_all_function(name=name, reload=reload)
 
         self._fs.makedirs("graphs", exist_ok=True)
-        dag.save(os.path.join(self._base_dir, f"graphs/{name}.{format}"))
+        dag.render(
+            os.path.join(self._base_dir, f"graphs/{name}"), format=format, cleanup=True
+        )
         rich.print(
             f"📊 Saved graph for {name} to {self._base_dir}/graphs/{name}.{format}"
         )
@@ -1132,7 +1134,8 @@ def show_dag(
         raw: bool = False,
     ):
         """
-        Display the graph of functions for a given name.
+        Display the graph of functions for a given name. By choosing the `raw` option, the graph object is returned.
+        The choosen format defines, which application is used to display the graph.
 
         Args:
             name (str): The name of the graph.