diff --git a/src/rai_bench/pyproject.toml b/src/rai_bench/pyproject.toml
index f892b88c8..90d820982 100644
--- a/src/rai_bench/pyproject.toml
+++ b/src/rai_bench/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rai-bench"
-version = "0.1.1"
+version = "0.1.2"
 description = "Package for running and creating benchmarks."
 authors = ["Jakub Matejczyk <jakub.matejczyk@robotec.ai>", "Magdalena Kotynia <magdalena.kotynia@robotec.ai>"]
 readme = "README.md"
diff --git a/src/rai_bench/rai_bench/examples/tool_calling_agent.py b/src/rai_bench/rai_bench/examples/tool_calling_agent.py
index 321ca9fbb..e76c18f00 100644
--- a/src/rai_bench/rai_bench/examples/tool_calling_agent.py
+++ b/src/rai_bench/rai_bench/examples/tool_calling_agent.py
@@ -45,7 +45,7 @@
 
     run_benchmark(
         llm=llm,
-        out_dir=args.out_dir,
+        out_dir=experiment_dir,
         tasks=tasks,
         bench_logger=bench_logger,
     )