diff --git a/test_wholeimage_swapmutil.py b/test_wholeimage_swapmutil.py
index 5f39642..89048ad 100644
--- a/test_wholeimage_swapmutil.py
+++ b/test_wholeimage_swapmutil.py
@@ -73,7 +73,8 @@ if __name__ == '__main__':
         swap_result = model(None, b_align_crop_tenor, latend_id, None, True)[0]
         swap_result_list.append(swap_result)
 
-    reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapmutil.jpg'))
+
+    reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapmutil.jpg'),opt.no_simswaplogo)
     print(' ')
 
     print('************ Done ! ************')
diff --git a/test_wholeimage_swapsingle.py b/test_wholeimage_swapsingle.py
index edf2816..0e3fb02 100644
--- a/test_wholeimage_swapsingle.py
+++ b/test_wholeimage_swapsingle.py
@@ -72,7 +72,7 @@ if __name__ == '__main__':
         swap_result = model(None, b_align_crop_tenor, latend_id, None, True)[0]
         swap_result_list.append(swap_result)
 
-    reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapsingle.jpg'))
+    reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapsingle.jpg'),opt.no_simswaplogo)
 
     print(' ')