From 73227677ea9e89c8861ca5cdce59b6c2896d4bcc Mon Sep 17 00:00:00 2001
From: rian <rian.hughes@physics.ox.ac.uk>
Date: Wed, 2 Oct 2024 16:38:30 +0300
Subject: [PATCH] fix additional test for TestStore

---
 blockchain/blockchain.go | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/blockchain/blockchain.go b/blockchain/blockchain.go
index 61597674de..46effbf072 100644
--- a/blockchain/blockchain.go
+++ b/blockchain/blockchain.go
@@ -653,11 +653,18 @@ func stateUpdateByHash(txn db.Transaction, hash *felt.Felt) (*core.StateUpdate,
 }
 
 func l1MsgHashes(txn db.Transaction, l1HandlerMsgHash *common.Hash) (*felt.Felt, error) {
-	var l1HandlerTxnHash *felt.Felt
+	var raw []byte
 	err := txn.Get(db.L1HandlerTxnHash.Key(l1HandlerMsgHash.Bytes()), func(val []byte) error {
-		return encoder.Unmarshal(val, l1HandlerTxnHash)
+		if len(val) == 0 {
+			return db.ErrKeyNotFound
+		}
+		raw = val
+		return nil
 	})
-	return l1HandlerTxnHash, err
+	if err != nil {
+		return nil, err
+	}
+	return new(felt.Felt).SetBytes(raw), nil
 }
 
 // SanityCheckNewHeight checks integrity of a block and resulting state update