Populate extraCoinSource field in `migrationPlanToSelectionWithdraw…

…als`.
cardano-foundation · May 18, 2021 · f9be46c · f9be46c
1 parent 83ee3f1
commit f9be46c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/core/src/Cardano/Wallet.hs b/lib/core/src/Cardano/Wallet.hs
@@ -1873,10 +1873,15 @@ migrationPlanToSelectionWithdrawals plan rewardWithdrawal outputAddressesToCycle
             { inputsSelected = view #inputIds migrationSelection
             , outputsCovered
             , utxoRemaining = UTxOIndex.empty
-            , extraCoinSource = Nothing
+            , extraCoinSource
             , changeGenerated = []
             }
 
+        extraCoinSource =
+            if (view #rewardWithdrawal migrationSelection) > Coin 0
+            then Just (view #rewardWithdrawal migrationSelection)
+            else Nothing
+
         withdrawal =
             if (view #rewardWithdrawal migrationSelection) > Coin 0
             then rewardWithdrawal