diff --git a/lora_diffusion/dataset.py b/lora_diffusion/dataset.py
index eed4403..fa99ac1 100644
--- a/lora_diffusion/dataset.py
+++ b/lora_diffusion/dataset.py
@@ -267,7 +267,7 @@ def __getitem__(self, index):
                     Image.open(self.mask_path[index % self.num_instance_images])
                 )
                 * 0.5
-                + 1
+                + 0.5
             )
 
         if self.h_flip and random.random() > 0.5: