swap_state: local_params getter missing RefundSig and Corearb

farcaster-project · Jun 21, 2022 · e9f6946 · e9f6946
1 parent 304a49b
commit e9f6946
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/swapd/swap_state.rs b/src/swapd/swap_state.rs
@@ -190,7 +190,9 @@ impl State {
             State::Alice(AliceState::CommitA { local_params, .. })
             | State::Bob(BobState::CommitB { local_params, .. })
             | State::Alice(AliceState::RevealA { local_params, .. })
-            | State::Bob(BobState::RevealB { local_params, .. }) => Some(local_params),
+            | State::Alice(AliceState::RefundSigA { local_params, .. })
+            | State::Bob(BobState::RevealB { local_params, .. })
+            | State::Bob(BobState::CorearbB { local_params, .. }) => Some(local_params),
             _ => None,
         }
     }