fix: node type bit

Signed-off-by: Zander Franks <[email protected]>
monadicus · May 17, 2024 · 356d293 · 356d293
1 parent 1f01c0b
commit 356d293
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/crates/snops-common/src/state/node_type.rs b/crates/snops-common/src/state/node_type.rs
@@ -1,6 +1,6 @@
 use std::ffi::OsStr;
 
-use crate::format::DataFormat;
+use crate::{format::DataFormat, prelude::MaskBit};
 
 #[derive(Debug, Clone, Copy, Hash, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
 #[serde(rename_all = "lowercase")]
@@ -36,7 +36,11 @@ impl NodeType {
     }
 
     pub fn bit(self) -> usize {
-        self as usize
+        (match self {
+            Self::Validator => MaskBit::Validator,
+            Self::Prover => MaskBit::Prover,
+            Self::Client => MaskBit::Client,
+        }) as usize
     }
 }