diff --git a/script/get-dataset-igbh/customize.py b/script/get-dataset-igbh/customize.py
index cacf4faa2..70b97a440 100644
--- a/script/get-dataset-igbh/customize.py
+++ b/script/get-dataset-igbh/customize.py
@@ -43,7 +43,7 @@ def preprocess(i):
     run_cmd += x_sep + \
         f"""{
             env['MLC_PYTHON_BIN_WITH_PATH']} tools/split_seeds.py --path {download_loc} --dataset_size {
-            env['MLC_DATASET_IGBH_SIZE']} {env.get('MLC_IGBH_IGBH_CALIBRATION_FLAG', '')} """
+            env['MLC_DATASET_IGBH_SIZE']} {env.get('MLC_IGBH_CALIBRATION_FLAG', '')} """
 
     # compress graph(for glt implementation)
     if env.get('MLC_IGBH_GRAPH_COMPRESS', '') == "yes":
diff --git a/script/get-dataset-igbh/meta.yaml b/script/get-dataset-igbh/meta.yaml
index d05bec5bf..b48e7b42c 100644
--- a/script/get-dataset-igbh/meta.yaml
+++ b/script/get-dataset-igbh/meta.yaml
@@ -439,7 +439,7 @@ variations:
       MLC_IGBH_GRAPH_COMPRESS: yes
   calibration:
     env:
-      MLC_IGBH_IGBH_CALIBRATION_FLAG: --calibration
+      MLC_IGBH_CALIBRATION_FLAG: --calibration
   csc:
     group: compressed-layout
     default: true