From 2ca8491f764b5ba25eefcd25edd25e7e9f20e247 Mon Sep 17 00:00:00 2001
From: Cassandra Heart <cassandra@quilibrium.com>
Date: Sun, 20 Oct 2024 20:46:36 -0500
Subject: [PATCH] b2-fixed

---
 node/consensus/data/pre_midnight_proof_worker.go | 10 ++++++++++
 node/store/coin.go                               | 11 ++++++++++-
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/node/consensus/data/pre_midnight_proof_worker.go b/node/consensus/data/pre_midnight_proof_worker.go
index 1ca6afa..c179671 100644
--- a/node/consensus/data/pre_midnight_proof_worker.go
+++ b/node/consensus/data/pre_midnight_proof_worker.go
@@ -6,6 +6,7 @@ import (
 
 	"github.com/iden3/go-iden3-crypto/poseidon"
 	"github.com/pkg/errors"
+	"source.quilibrium.com/quilibrium/monorepo/node/consensus"
 	"source.quilibrium.com/quilibrium/monorepo/node/execution/intrinsics/token/application"
 	"source.quilibrium.com/quilibrium/monorepo/node/protobufs"
 	"source.quilibrium.com/quilibrium/monorepo/node/store"
@@ -24,6 +25,15 @@ func (e *DataClockConsensusEngine) runPreMidnightProofWorker() {
 		panic(err)
 	}
 
+	for {
+		if e.state < consensus.EngineStateCollecting {
+			e.logger.Debug("waiting for node to finish starting")
+			time.Sleep(10 * time.Second)
+			continue
+		}
+		break
+	}
+
 	addrBI, err := poseidon.HashBytes(e.pubSub.GetPeerID())
 	if err != nil {
 		panic(err)
diff --git a/node/store/coin.go b/node/store/coin.go
index 58f99b7..f3fe0a9 100644
--- a/node/store/coin.go
+++ b/node/store/coin.go
@@ -480,7 +480,16 @@ func (p *PebbleCoinStore) Migrate(filter []byte) error {
 				panic(err)
 			}
 
-			return nil
+			txn, err := p.NewTransaction()
+			if err != nil {
+				return nil
+			}
+
+			err = txn.Set(migrationKey(), []byte{0x02, 0x00, 0x01, 0x02})
+			if err != nil {
+				panic(err)
+			}
+			return txn.Commit()
 		}
 		return nil
 	}