From 33f9242d19f778d8730d60a080d8e5d9ca03cb42 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Toni=20Ram=C3=ADrez?=
 <58293609+ToniRamirezM@users.noreply.github.com>
Date: Mon, 12 Aug 2024 13:28:25 +0200
Subject: [PATCH] fix error check (#110)

---
 aggregator/aggregator.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aggregator/aggregator.go b/aggregator/aggregator.go
index f3598a5..21df459 100644
--- a/aggregator/aggregator.go
+++ b/aggregator/aggregator.go
@@ -1284,7 +1284,7 @@ func (a *Aggregator) tryGenerateBatchProof(ctx context.Context, prover proverInt
 	log.Debug("tryGenerateBatchProof start")
 
 	batchToProve, witness, proof, err0 := a.getAndLockBatchToProve(ctx, prover)
-	if errors.Is(err0, state.ErrNotFound) {
+	if errors.Is(err0, state.ErrNotFound) || errors.Is(err0, entities.ErrNotFound) {
 		// nothing to proof, swallow the error
 		log.Debug("Nothing to generate proof")
 		return false, nil