diff --git a/SDFMapCreator/Program.cs b/SDFMapCreator/Program.cs
index 52e9051..8320b5c 100644
--- a/SDFMapCreator/Program.cs
+++ b/SDFMapCreator/Program.cs
@@ -115,7 +115,7 @@ public static class Program {
             if (i >= Images.Count - 1) continue;
             
             var mask = GetABMask(Images[i].Pixels, Images[i + 1].Pixels);
-            TransitionMasks.Add(new(mask, Images[i], Images[i + 1]));
+            TransitionMasks.Add(new(mask));
         }
 
         //EdgeDetect all masks
diff --git a/SDFMapCreator/Records.cs b/SDFMapCreator/Records.cs
index aabbed0..da977c9 100644
--- a/SDFMapCreator/Records.cs
+++ b/SDFMapCreator/Records.cs
@@ -11,4 +11,4 @@ public record MaskData(Vector3[,] Mask, Image Image, List<Vector2> Edges) {
 
 public record SDFData(Vector3[,] SDF);
 
-public record TransitionMaskData(Vector3[,] Mask, Image ImageA, Image ImageB);
\ No newline at end of file
+public record TransitionMaskData(Vector3[,] Mask);
\ No newline at end of file