diff --git a/stable_audio_tools/inference/generation.py b/stable_audio_tools/inference/generation.py
index 843ab4b..76b8861 100644
--- a/stable_audio_tools/inference/generation.py
+++ b/stable_audio_tools/inference/generation.py
@@ -270,5 +270,6 @@ def build_mask(sample_size, mask_args):
     # marination finishes the inpainting early in the denoising schedule, and lets audio get changed in the final rounds
     if marination > 0:        
         mask = mask * (1-marination) 
+    mask = 1 - mask
     #print(mask)
     return mask
diff --git a/stable_audio_tools/inference/sampling.py b/stable_audio_tools/inference/sampling.py
index 2229e50..4378c32 100644
--- a/stable_audio_tools/inference/sampling.py
+++ b/stable_audio_tools/inference/sampling.py
@@ -90,7 +90,7 @@ def sample(model, x, steps, eta, **extra_args):
 def get_bmask(i, steps, mask):
     strength = (i+1)/(steps)
     # convert to binary mask
-    bmask = torch.where(mask<=strength,1,0)
+    bmask = torch.where(mask<strength,0,1)
     return bmask
 
 def make_cond_model_fn(model, cond_fn):