diff --git a/ea-scripts/Auto Layout.md b/ea-scripts/Auto Layout.md
index 268646e..38af32c 100644
--- a/ea-scripts/Auto Layout.md	
+++ b/ea-scripts/Auto Layout.md	
@@ -308,16 +308,8 @@ function loadELK(doAfterLoaded) {
 function normalizeSelectedArrows() {
   let gapValue = 2;
 
-  const selectedIndividualArrows = ea
-    .getMaximumGroups(ea.getViewSelectedElements())
-    .reduce(
-      (result, group) =>
-        group.length === 1 &&
-        (group[0].type === "arrow" || group[0].type === "line")
-          ? [...result, group[0]]
-          : result,
-      []
-    );
+  const selectedIndividualArrows = ea.getMaximumGroups(ea.getViewSelectedElements())
+    .reduce((result, g) => [...result, ...g.filter(el => el.type === 'arrow')], []);
 
   const allElements = ea.getViewElements();
   for (const arrow of selectedIndividualArrows) {
diff --git a/ea-scripts/Normalize Selected Arrows.md b/ea-scripts/Normalize Selected Arrows.md
index 573e75b..ce4b8af 100644
--- a/ea-scripts/Normalize Selected Arrows.md	
+++ b/ea-scripts/Normalize Selected Arrows.md	
@@ -26,8 +26,7 @@ if(!settings["Gap"]) {
 let gapValue = settings["Gap"].value;
 
 const selectedIndividualArrows = ea.getMaximumGroups(ea.getViewSelectedElements())
-	.reduce((result, group) => (group.length === 1 && (group[0].type === 'arrow' || group[0].type === 'line')) ? 
-			[...result, group[0]] : result, []);
+    .reduce((result, g) => [...result, ...g.filter(el => el.type === 'arrow')], []);
 
 const allElements = ea.getViewElements();
 for(const arrow of selectedIndividualArrows) {