diff --git a/modules_forge/unet_patcher.py b/modules_forge/unet_patcher.py
index a20ccb70..4c00fa84 100644
--- a/modules_forge/unet_patcher.py
+++ b/modules_forge/unet_patcher.py
@@ -102,7 +102,7 @@ class UnetPatcher(ModelPatcher):
         self.append_model_option('conditioning_modifiers', modifier, ensure_uniqueness)
         return
 
-    def add_memory_peak_estimation_modifier(self, modifier):
+    def set_memory_peak_estimation_modifier(self, modifier):
         self.model_options['memory_peak_estimation_modifier'] = modifier
         return