diff --git a/model/STG-correction/utils/data_utils.py b/model/STG-correction/utils/data_utils.py
index 9674e34..c13c0b4 100644
--- a/model/STG-correction/utils/data_utils.py
+++ b/model/STG-correction/utils/data_utils.py
@@ -425,13 +425,15 @@ def convert_spmap_sw(sp_map : list, sw_labels : np.array) -> list:
 def convert_spmap_tg(sp_map : list, tg_mapper : list) -> list:
     new_spmap = []
     for idx in range(len(sp_map)):
+        if idx == 163:
+            print()
         spmapper = sp_map[idx]
         tgmapper = tg_mapper[idx]
         mapper = {}
         if not spmapper: new_spmap.append(None)
         else:
             for sp in spmapper:
-                if sp in tgmapper and tgmapper[sp] > 0: mapper[tgmapper[sp]] = spmapper[sp]
+                if sp in tgmapper and tgmapper[sp] >= 0: mapper[tgmapper[sp]] = spmapper[sp]
             new_spmap.append(mapper)
     return new_spmap