From d529a1b6e41a7860f11271eb3ccf9123731e1e0e Mon Sep 17 00:00:00 2001
From: Joshua Kim <20001595+joshua-kim@users.noreply.github.com>
Date: Thu, 4 Apr 2024 03:18:18 -0400
Subject: [PATCH] fix

Signed-off-by: Joshua Kim <20001595+joshua-kim@users.noreply.github.com>
---
 vms/proposervm/state/state.go      | 4 ++++
 vms/proposervm/state/state_test.go | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/vms/proposervm/state/state.go b/vms/proposervm/state/state.go
index bed817db02de..9ec7a6cd46fc 100644
--- a/vms/proposervm/state/state.go
+++ b/vms/proposervm/state/state.go
@@ -108,6 +108,10 @@ func (s *state) pruneVerifiedBlocks(db *versiondb.Database) error {
 		if err := s.chainState.DeleteVerifiedBlock(blkID); err != nil {
 			return nil
 		}
+
+		if err := s.BlockState.DeleteBlock(blkID); err != nil {
+			return nil
+		}
 	}
 
 	if err := iter.Error(); err != nil {
diff --git a/vms/proposervm/state/state_test.go b/vms/proposervm/state/state_test.go
index a69b0647ab22..7bee5dd86880 100644
--- a/vms/proposervm/state/state_test.go
+++ b/vms/proposervm/state/state_test.go
@@ -10,6 +10,7 @@ import (
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/stretchr/testify/require"
 
+	"github.com/ava-labs/avalanchego/database"
 	"github.com/ava-labs/avalanchego/database/memdb"
 	"github.com/ava-labs/avalanchego/database/versiondb"
 	"github.com/ava-labs/avalanchego/ids"
@@ -125,6 +126,9 @@ func TestPruneVerifiedBlocksOnRestart(t *testing.T) {
 				ok, err := state.HasVerifiedBlock(blk.ID())
 				require.NoError(err)
 				require.False(ok)
+
+				_, _, err = state.GetBlock(blk.ID())
+				require.ErrorIs(err, database.ErrNotFound)
 			}
 
 			for _, blk := range tt.preferredBlks {