diff --git a/src/init/features.py b/src/init/features.py
index 570c1c0..017d4fe 100644
--- a/src/init/features.py
+++ b/src/init/features.py
@@ -87,7 +87,7 @@ def decorate_2d(
 
     inp_normals = normals_from_depth_sobel(
         inp_depth.cpu().numpy(),
-        bg_mask=inp_mask.squeeze().cpu().numpy(),
+        bg_mask=~inp_mask.squeeze().cpu().numpy(),
     )
 
     # * Mask out background