diff --git a/LayoutFunctions/LayoutFunctionCommon/WallGeneration.cs b/LayoutFunctions/LayoutFunctionCommon/WallGeneration.cs
index e43f22e0..6b0fd507 100644
--- a/LayoutFunctions/LayoutFunctionCommon/WallGeneration.cs
+++ b/LayoutFunctions/LayoutFunctionCommon/WallGeneration.cs
@@ -272,7 +272,13 @@ public static List<RoomEdge> SplitOverlappingWallCandidates(IEnumerable<RoomEdge
                     resultCandidates.Add(candidate);
                     continue;
                 }
-                var linesOrderedByLength = collinearLinesGroup.Value.Where(x => !x.Item2.Contains("Partition")).OrderByDescending(v => v.Line.Length());
+                var filteredLines = collinearLinesGroup.Value.Where(x => !x.Item2.Contains("Partition"));
+                if (!filteredLines.Any())
+                {
+                    filteredLines = collinearLinesGroup.Value;
+                }
+
+                var linesOrderedByLength = filteredLines.OrderByDescending(v => v.Line.Length());
 
                 var dominantLineForGroup = linesOrderedByLength.FirstOrDefault(x => x.PrimaryEntryEdge == true);