diff --git a/Tutorial_0.ipynb b/Tutorial_0.ipynb
index 4cc965f..6886fa7 100644
--- a/Tutorial_0.ipynb
+++ b/Tutorial_0.ipynb
@@ -220,11 +220,12 @@
     "dataset = dl_info['dataset']\n",
     "seq_length, feature_dim = dataset.window, dataset.var_num\n",
     "ori_data = np.load(os.path.join(dataset.dir, f\"sine_ground_truth_{seq_length}_train.npy\"))\n",
-    "# ori_data = np.load(os.path.join(dataset.dir, f\"{dataset_name}_norm_truth_{seq_length}_train.npy\"))  # Uncomment the line if dataset other than Sine is used.\n",
+    "# ori_data = np.load(os.path.join(dataset.dir, f\"{dataset.name}_norm_truth_{seq_length}_train.npy\"))  # Uncomment the line if dataset other than Sine is used.\n",
     "fake_data = trainer.sample(num=len(dataset), size_every=2001, shape=[seq_length, feature_dim])\n",
     "if dataset.auto_norm:\n",
     "    fake_data = unnormalize_to_zero_to_one(fake_data)\n",
-    "    np.save(os.path.join(args.save_dir, f'ddpm_fake_sines.npy'), fake_data)"
+    "    np.save(os.path.join(args.save_dir, f'ddpm_fake_sines.npy'), fake_data)\n",
+    "    # np.save(os.path.join(args.save_dir, f'ddpm_fake_{dataset.name}.npy'), fake_data)  # Uncomment the line if dataset other than Sine is used."
    ]
   },
   {