From 9f1c9adc3b17edfd9ed9187e8184b4142bd22841 Mon Sep 17 00:00:00 2001
From: Johannes Lund <anviking@me.com>
Date: Wed, 13 Sep 2023 17:44:15 +0200
Subject: [PATCH] Re-add arbitrary pparam in shrink property

---
 .../Cardano/Wallet/Write/Tx/Balance/TokenBundleSizeSpec.hs   | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/lib/wallet/test/unit/Cardano/Wallet/Write/Tx/Balance/TokenBundleSizeSpec.hs b/lib/wallet/test/unit/Cardano/Wallet/Write/Tx/Balance/TokenBundleSizeSpec.hs
index 986eeb36437..a8a766c865b 100644
--- a/lib/wallet/test/unit/Cardano/Wallet/Write/Tx/Balance/TokenBundleSizeSpec.hs
+++ b/lib/wallet/test/unit/Cardano/Wallet/Write/Tx/Balance/TokenBundleSizeSpec.hs
@@ -108,8 +108,9 @@ prop_assessTokenBundleSize_enlarge b1' b2' pp =
 prop_assessTokenBundleSize_shrink
     :: Blind (VariableSize1024 TokenBundle)
     -> Blind (VariableSize16 TokenBundle)
+    -> PParamsInRecentEra
     -> Property
-prop_assessTokenBundleSize_shrink b1' b2' =
+prop_assessTokenBundleSize_shrink b1' b2' pp =
     assess b1 == TokenBundleSizeWithinLimit ==> conjoin
         [ assess (b1 `TokenBundle.difference` b2)
             === TokenBundleSizeWithinLimit
@@ -117,7 +118,7 @@ prop_assessTokenBundleSize_shrink b1' b2' =
             === TokenBundleSizeWithinLimit
         ]
   where
-    assess = assessTokenBundleSize babbageTokenBundleSizeAssessor
+    assess = assessTokenBundleSize $ mkAssessorFromPParamsInRecentEra pp
     b1 = unVariableSize1024 $ getBlind b1'
     b2 = unVariableSize16 $ getBlind b2'