Simplify implementation of TokenBundle.difference.

cardano-foundation · Jun 2, 2023 · a09b324 · a09b324
1 parent a200e7a
commit a09b324
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/primitive/lib/Cardano/Wallet/Primitive/Types/TokenBundle.hs b/lib/primitive/lib/Cardano/Wallet/Primitive/Types/TokenBundle.hs
@@ -95,6 +95,8 @@ import Data.Map.Strict
     ( Map )
 import Data.Monoid.Cancellative
     ( Reductive ((</>)) )
+import Data.Monoid.Monus
+    ( Monus ((<\>)) )
 import Data.Ord
     ( comparing )
 import Data.Set
@@ -309,9 +311,7 @@ subtract (TokenBundle c1 m1) (TokenBundle c2 m2) =
 --
 difference :: TokenBundle -> TokenBundle -> TokenBundle
 difference (TokenBundle c1 m1) (TokenBundle c2 m2) =
-    TokenBundle
-        (Coin.difference c1 c2)
-        (TokenMap.difference m1 m2)
+    TokenBundle (c1 <\> c2) (m1 <\> m2)
 
 --------------------------------------------------------------------------------
 -- Quantities