diff --git a/resources/profiles/BBL/machine/Bambu Lab H2S 0.4 nozzle.json b/resources/profiles/BBL/machine/Bambu Lab H2S 0.4 nozzle.json
index f86c4def40..4ceb89c7b3 100644
--- a/resources/profiles/BBL/machine/Bambu Lab H2S 0.4 nozzle.json	
+++ b/resources/profiles/BBL/machine/Bambu Lab H2S 0.4 nozzle.json	
@@ -54,6 +54,10 @@
         "32",
         "32"
     ],
+    "nozzle_flush_dataset": [
+        "1",
+        "2"
+    ],
     "printable_area": [
         "0x0",
         "340x0",
diff --git a/src/slic3r/Utils/CalibUtils.cpp b/src/slic3r/Utils/CalibUtils.cpp
index 4081d7b3a2..392a5661c5 100644
--- a/src/slic3r/Utils/CalibUtils.cpp
+++ b/src/slic3r/Utils/CalibUtils.cpp
@@ -866,6 +866,9 @@ void CalibUtils::set_for_auto_pa_model_and_config(const std::vector<CalibInfo> &
     // nozzle volume type
     std::vector<int>& nozzle_volume_types = full_config.option<ConfigOptionEnumsGeneric>("nozzle_volume_type", true)->values;
     nozzle_volume_types.resize(extruder_count, NozzleVolumeType::nvtStandard);
+    auto nozzle_flush_dataset = full_config.option<ConfigOptionIntsNullable>("nozzle_flush_dataset", true)->values;
+    nozzle_flush_dataset.resize(extruder_count, 0);
+
     int               filament_nums = calib_infos.size();
     std::vector<int> physical_extruder_maps = dynamic_cast<ConfigOptionInts *>(full_config.option("physical_extruder_map", true))->values;
     for (size_t filament_index = 0; filament_index < calib_infos.size(); ++filament_index) {
@@ -899,7 +902,7 @@ void CalibUtils::set_for_auto_pa_model_and_config(const std::vector<CalibInfo> &
                     flush_matrix_vec.emplace_back(0);
                 }
                 else {
-                    Slic3r::FlushVolCalculator calculator(min_flush_volumes[from_idx], Slic3r::g_max_flush_volume, extruder_count > 1, NozzleVolumeType(nozzle_volume_types[e_idx]));
+                    Slic3r::FlushVolCalculator calculator(min_flush_volumes[from_idx], Slic3r::g_max_flush_volume, nozzle_flush_dataset[e_idx]);
                     wxColour from = wxColour(filament_colors[from_idx]);
                     wxColour to = wxColour(filament_colors[to_idx]);
                     int volume = calculator.calc_flush_vol(from.Alpha(), from.Red(), from.Green(), from.Blue(), to.Alpha(), to.Red(), to.Green(), to.Blue());