File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -35,13 +35,11 @@ import Contract.Wallet
3535 ( getWalletAddresses
3636 , getWalletBalance
3737 , ownPaymentPubKeyHashes
38+ , ownStakePubKeyHashes
3839 , privateKeysToKeyWallet
3940 , withKeyWallet
4041 )
41- import Contract.Wallet.Key
42- ( getPrivatePaymentKey
43- , getPrivateStakeKey
44- )
42+ import Contract.Wallet.Key (getPrivatePaymentKey , getPrivateStakeKey )
4543import Contract.Wallet.KeyFile
4644 ( privatePaymentKeyFromTextEnvelope
4745 , privatePaymentKeyToFile
@@ -386,9 +384,13 @@ returnFunds backup env allWalletsArray mbFundTotal hasRun =
386384 pkhs <- fold <$> for nonEmptyWallets
387385 (snd >>> flip withKeyWallet ownPaymentPubKeyHashes)
388386
387+ skhs <- catMaybes <<< fold <$> for nonEmptyWallets
388+ (snd >>> flip withKeyWallet ownStakePubKeyHashes)
389+
389390 let
390391 constraints = flip foldMap (Map .keys utxos) mustSpendPubKeyOutput
391392 <> foldMap mustBeSignedBy pkhs
393+ <> foldMap (mustBeSignedBy <<< wrap <<< unwrap) skhs
392394 lookups = unspentOutputs utxos
393395
394396 unbalancedTx /\ usedUtxos <- liftedE $ mkUnbalancedTxImpl lookups
You can’t perform that action at this time.
0 commit comments