safer state checking

anoma · Mar 15, 2024 · cce1c9a · cce1c9a
1 parent 5790f75
commit cce1c9a
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/crates/sdk/src/tx.rs b/crates/sdk/src/tx.rs
@@ -1995,11 +1995,10 @@ pub async fn build_vote_proposal(
             voter_address,
             Some(current_epoch),
         )
-        .await?;
-        if matches!(
-            state,
-            Some(ValidatorState::Jailed) | Some(ValidatorState::Inactive)
-        ) {
+        .await?
+        .expect("Expected to find the state of the validator");
+
+        if matches!(state, ValidatorState::Jailed | ValidatorState::Inactive) {
             return Err(Error::from(TxSubmitError::CannotVoteInGovernance(
                 voter_address.clone(),
                 current_epoch,
@@ -2041,12 +2040,12 @@ pub async fn build_vote_proposal(
             voter_address,
             Some(current_epoch),
         )
-        .await
+        .await?
         .expect("Expected to find the state of the validator");
 
         if !matches!(
             val_state,
-            Some(ValidatorState::Jailed) | Some(ValidatorState::Inactive)
+            ValidatorState::Jailed | ValidatorState::Inactive
         ) {
             vec![voter_address.clone()]
         } else {