tighten code

1 files changed, 2 insertions, 5 deletions

diff --git a/src/dk.biks.bachelorizer/dk/biks/bachelorizer/Combi.java b/src/dk.biks.bachelorizer/dk/biks/bachelorizer/Combi.java
index 8bbe5ef..82f9cc6 100644
--- a/src/dk.biks.bachelorizer/dk/biks/bachelorizer/Combi.java
+++ b/src/dk.biks.bachelorizer/dk/biks/bachelorizer/Combi.java
@@ -154,12 +154,9 @@ public final class Combi {
 
 			// add set of current and any unconnected modules
 			Set<Vertex> isolated = all_set.stream().filter(x -> {
-				if (x == current)
-					return false;
-				Integer weight = g.getWeight(current, x);
-				return weight == null && !done.contains(x);
+				return x == current
+					|| g.getWeight(current, x) == null;
 			}).collect(Collectors.toSet());
-			isolated.add(current);
 
 			// add as set and omit from future iterations
 			sets.add(isolated);