From 75b533228893ded2cecf83b7eb87d34b50b709c2 Mon Sep 17 00:00:00 2001
From: solidsnakedev <jona.ca.eng@gmail.com>
Date: Tue, 17 Sep 2024 19:45:39 -0600
Subject: [PATCH] fix: Add wrap constructor to V1/V2 stake_credential when
 Purpose is Rewarding

---
 crates/uplc/src/tx/to_plutus_data.rs | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/crates/uplc/src/tx/to_plutus_data.rs b/crates/uplc/src/tx/to_plutus_data.rs
index aab608809..d3d7aaae9 100644
--- a/crates/uplc/src/tx/to_plutus_data.rs
+++ b/crates/uplc/src/tx/to_plutus_data.rs
@@ -872,9 +872,10 @@ impl<'a> ToPlutusData for WithWrappedTransactionId<'a, ScriptPurpose> {
             ScriptPurpose::Spending(out_ref, ()) => {
                 wrap_with_constr(1, WithWrappedTransactionId(out_ref).to_plutus_data())
             }
-            ScriptPurpose::Rewarding(stake_credential) => {
-                wrap_with_constr(2, stake_credential.to_plutus_data())
-            }
+            ScriptPurpose::Rewarding(stake_credential) => wrap_with_constr(
+                2,
+                WithWrappedStakeCredential(stake_credential).to_plutus_data(),
+            ),
             ScriptPurpose::Certifying(_, dcert) => {
                 wrap_with_constr(3, WithPartialCertificates(dcert).to_plutus_data())
             }
@@ -1243,6 +1244,12 @@ impl<'a> ToPlutusData for WithWrappedStakeCredential<'a, KeyValuePairs<Address,
     }
 }
 
+impl<'a> ToPlutusData for WithWrappedStakeCredential<'a, StakeCredential> {
+    fn to_plutus_data(&self) -> PlutusData {
+        wrap_with_constr(0, self.0.to_plutus_data())
+    }
+}
+
 impl ToPlutusData for Voter {
     fn to_plutus_data(&self) -> PlutusData {
         match self {