diff --git a/data/generate_dataset.py b/data/generate_dataset.py
index 4f9f793..76d1273 100644
--- a/data/generate_dataset.py
+++ b/data/generate_dataset.py
@@ -104,7 +104,7 @@ def main(args):
             save_spec_pic(args.folder, s_trans, times, freq, t_idx0, t_idx1, f_idx0, f_idx1, t_res, f_res)
             exit()
         else:
-            fig_title = (f'{Path(folder).name}__{t0:.0f}s-{t1:.0f}s__{f0:4.0f}-{f1:4.0f}Hz').replace(' ', '0')
+            fig_title = (f'{Path(args.folder).name}__{t0:.0f}s-{t1:.0f}s__{f0:4.0f}-{f1:4.0f}Hz').replace(' ', '0')
             fig = plt.figure(figsize=(7, 7), num=fig_title)
             gs = gridspec.GridSpec(1, 2, width_ratios=(8, 1), wspace=0)  # , bottom=0, left=0, right=1, top=1
             ax = fig.add_subplot(gs[0, 0])