diff --git a/codes/ecg-image-generator/helper_functions.py b/codes/ecg-image-generator/helper_functions.py
index 26e5a41..6f660c6 100644
--- a/codes/ecg-image-generator/helper_functions.py
+++ b/codes/ecg-image-generator/helper_functions.py
@@ -344,13 +344,7 @@ def write_wfdb_file(ecg_frame, filename, rate, header_file, write_dir, full_mode
                 sig_name = leads, p_signal = array.T, fmt = header.fmt,
                 adc_gain = header.adc_gain, baseline = header.baseline, 
                 base_time = header.base_time, base_date = header.base_date, 
-                write_dir = write_dir)
-
-    with open(os.path.join(write_dir, tail + '.hea'), "a") as f:
-        for line in header.comments:
-            f.write("#" + line)
-            f.write("\n")
-
+                write_dir = write_dir, comments = header.comments)
 
 def get_lead_pixel_coordinate(leads):