From 8546a183e6be891fae3019b256d8c18e5cd0ef23 Mon Sep 17 00:00:00 2001
From: jonesmarvin8 <83104039+jonesmarvin8@users.noreply.github.com>
Date: Wed, 1 Apr 2026 16:39:33 -0400
Subject: [PATCH] fixed tests

---
 lee/state_machine/src/signature/private_key.rs | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/lee/state_machine/src/signature/private_key.rs b/lee/state_machine/src/signature/private_key.rs
index f1a5398a..1ff185aa 100644
--- a/lee/state_machine/src/signature/private_key.rs
+++ b/lee/state_machine/src/signature/private_key.rs
@@ -68,15 +68,14 @@ impl PrivateKey {
             return Err(LeeError::InvalidPrivateKey);
         }
 
-        let sk = k256::SecretKey::from_bytes(value.into()).expect("Expect a valid secret key");
-        let pk = sk.public_key();
+        let sk = k256::SecretKey::from_slice(value).expect("Expect a valid secret key");
 
-        let hashed: [u8; 32] = Sha256::digest(pk.to_encoded_point(true).as_bytes()).into();
+        let hashed: [u8; 32] = Sha256::digest(sk.public_key().to_encoded_point(true).as_bytes()).into();
+
+        let sk = sk.to_nonzero_scalar();
 
         Self::try_new(
-            k256::Scalar::from_repr((*value).into())
-                .expect("Expect a valid k256 scalar")
-                .add(&k256::Scalar::from_repr(hashed.into()).expect("Expect a valid k256 scalar"))
+            sk.add(&k256::Scalar::from_repr(hashed.into()).expect("Expect a valid k256 scalar"))
                 .to_bytes()
                 .into(),
         )