diff --git a/func/Module_bcg_quality_label.py b/func/Module_bcg_quality_label.py
index b524db7..d903e9f 100644
--- a/func/Module_bcg_quality_label.py
+++ b/func/Module_bcg_quality_label.py
@@ -1030,7 +1030,16 @@ class Data():
                 artifact_end = append(artifact_end, self.Artifact_a[i])
             self.artifact_mask = zeros(len(self.BCG))
             for i in range(0, len(self.artifact_number)):
-                self.artifact_mask[artifact_start[i]: artifact_end[i] + 1] = 1
+                if self.artifact_type[i] == 1:
+                    self.artifact_mask[artifact_start[i]: artifact_end[i] + 1] = 1
+                elif self.artifact_type[i] == 2:
+                    self.artifact_mask[artifact_start[i]: artifact_end[i] + 1] = 2
+                elif self.artifact_type[i] == 3:
+                    self.artifact_mask[artifact_start[i]: artifact_end[i] + 1] = 3
+                elif self.artifact_type[i] == 4:
+                    self.artifact_mask[artifact_start[i]: artifact_end[i] + 1] = 4
+                elif self.artifact_type[i] == 5:
+                    self.artifact_mask[artifact_start[i]: artifact_end[i] + 1] = 5
         except Exception as e:
             return Result().failure(info=Constants.INPUT_FAILURE +
                                          Constants.FAILURE_REASON["Get_Artifact_Format_Exception"] + "\n" + format_exc())