diff --git a/pyalfe/tasks/segmentation.py b/pyalfe/tasks/segmentation.py
index 5962976..91c62dd 100644
--- a/pyalfe/tasks/segmentation.py
+++ b/pyalfe/tasks/segmentation.py
@@ -419,8 +419,8 @@ def post_process(self, pred, seg, mask=None):
                 26: 4,
                 28: 4,
                 49: 4,
-                50: 41,
-                51: 41,
+                50: 4,
+                51: 4,
                 52: 4,
                 54: 4,
                 58: 4,