diff --git a/matlab/tfgm/datasets/get_cuicui_mix_icassp.m b/matlab/tfgm/datasets/get_cuicui_mix_icassp.m
index 956c5895d84c3ce60297d2f0371d7186b286ed7a..878f6837ce86b1bef04882783a370178714a42a0 100644
--- a/matlab/tfgm/datasets/get_cuicui_mix_icassp.m
+++ b/matlab/tfgm/datasets/get_cuicui_mix_icassp.m
@@ -75,32 +75,31 @@ figure; imagesc(mask);
 
 %%  plot figures
 
-pwd;
-pathname ='figures';
-if ~exist('figures','dir')
-    mkdir('figures');
+fig_dir ='figures_car_bird_icassp';
+if ~exist(fig_dir,'dir')
+    mkdir(fig_dir);
 end
-addpath('figures')
+addpath(fig_dir)
+
 
 
 figure;
 plot_spectrogram(mask, dgt_params, signal_params, dgt);
 title('mask')
-saveas(gcf,fullfile(pathname,['mask','.png']));
+saveas(gcf,fullfile(fig_dir,'mask.png'));
 
 figure;
 plot_spectrogram(dgt_Xref, dgt_params, signal_params, dgt)
 title('target signal')
-saveas(gcf,fullfile(pathname,['engine','.png']));
+saveas(gcf,fullfile(fig_dir,'engine.png'));
 figure;
 plot_spectrogram(dgt_Xper, dgt_params, signal_params, dgt)
 title('perturbation signal')
-saveas(gcf,fullfile(pathname,['bird','.png']));
+saveas(gcf,fullfile(fig_dir,'bird.png'));
 figure;
 plot_spectrogram(dgt_mix, dgt_params, signal_params, dgt)
 title('perturbation signal')
-saveas(gcf,fullfile(pathname,['engine_bird','.png']));
-
+saveas(gcf,fullfile(fig_dir,'engine_bird.png'));