diff --git a/library/train_util.py b/library/train_util.py
index 200b486c..35bfb5f5 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1705,6 +1705,8 @@ class ControlNetDataset(BaseDataset):
                 subset.caption_dropout_rate,
                 subset.caption_dropout_every_n_epochs,
                 subset.caption_tag_dropout_rate,
+                subset.caption_prefix,
+                subset.caption_suffix,
                 subset.token_warmup_min,
                 subset.token_warmup_step,
             )