diff --git a/csst_cpic_sim/main.py b/csst_cpic_sim/main.py
index 7b6fd01b2741504a929abcf3219f7958f0c61291..a2e27de43b9f4f964d4b887aefbbc25b2f7cc3be 100644
--- a/csst_cpic_sim/main.py
+++ b/csst_cpic_sim/main.py
@@ -371,6 +371,7 @@ def observation_simulation_from_config(obs_file, config_file):
             time = obs_info.get('time', 0)
             emgain = obs_info.get('emgain', None)
             time = relative_time(time)
+            dataset = obs_info.get('dataset', 'default')
 
         except Exception as e:
             log.error(f"{file} is not a valid yaml file.")
@@ -416,6 +417,7 @@ def observation_simulation_from_config(obs_file, config_file):
                     output=output,
                     nsample=nsample,
                     csst_format=csst_format,
+                    dataset=dataset,
                     prograss_bar=True)
             except Exception as e:
                 log.error(f"{info_text} failed with {type(e).__name__}{e}.\n\n {traceback.format_exc()}")
diff --git a/example/observations/05_sci.yaml b/example/observations/05_sci.yaml
index b392407beeda5369aceba9bcec258b59e1ec914b..acec3371c2b77b0e437788bb75da1aaa54fc1ee3 100644
--- a/example/observations/05_sci.yaml
+++ b/example/observations/05_sci.yaml
@@ -5,6 +5,7 @@ expt: 100
 nframe: 2
 obsid: '40100000005'
 rotation: 30
+dataset: cpic-v01
 shift:
 - 0
 - 0