diff --git a/openfasoc/generators/temp-sense-gen/tools/parse_rpt.py b/openfasoc/generators/temp-sense-gen/tools/parse_rpt.py
index d0c40630c..b574d3c77 100644
--- a/openfasoc/generators/temp-sense-gen/tools/parse_rpt.py
+++ b/openfasoc/generators/temp-sense-gen/tools/parse_rpt.py
@@ -3,6 +3,12 @@
 sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..'))
 from common.check_gen_files import check_gen_files
 
+_generator_is = {
+    'sky130hvl_ldo': 0, 
+    'sky130hd_temp': 1, 
+    'sky130XX_cryo': 0
+}
+
 drc_filename = "work/6_final_drc.rpt"
 num_lines = sum(1 for line in open(drc_filename))
 
@@ -24,7 +30,7 @@
 
 json_filename = "test.json"
 
-if check_gen_files(json_filename, (len(sys.argv) == 1)):
+if check_gen_files(json_filename, _generator_is, " "):
         print("Flow check is clean!")
 else:
     print("Flow check failed!")