More effificent TSet difference

1 files changed, 4 insertions, 3 deletions

diff --git a/lib/pure/collections/sets.nim b/lib/pure/collections/sets.nim
index 2f0c4d3d2..f1eed0004 100644
--- a/lib/pure/collections/sets.nim
+++ b/lib/pure/collections/sets.nim
@@ -169,9 +169,10 @@ proc intersection*[A](s1, s2: TSet[A]): TSet[A] =
 
 proc difference*[A](s1, s2: TSet[A]): TSet[A] =
   ## returns a new set of all items that are contained in `s1`, but not in `s2`
-  result = s1
-  for item in s2:
-    if contains(result, item): excl(result, item)
+  result = initSet[A]()
+  for item in s1:
+    if not contains(s2, item):
+      incl(result, item)
 
 proc symmetricDifference*[A](s1, s2: TSet[A]): TSet[A] =
   ## returns a new set of all items that are contained in either