diff --git a/consensus/polybft/fsm_test.go b/consensus/polybft/fsm_test.go
index 1dddcb5e70..91c6bef968 100644
--- a/consensus/polybft/fsm_test.go
+++ b/consensus/polybft/fsm_test.go
@@ -586,6 +586,7 @@ func TestFSM_VerifyStateTransactions_DistributeRewards(t *testing.T) {
 			validators:             validatorSet,
 			distributeRewardsInput: createTestDistributeRewardsInput(t, 0, validators.GetPublicIdentities(), 10),
 			logger:                 hclog.NewNullLogger(),
+			forks:                  &chain.Forks{chain.Governance: chain.NewFork(0)},
 		}
 
 		distributeRewardsTx, err := fsm.createDistributeRewardsTx()
@@ -603,6 +604,7 @@ func TestFSM_VerifyStateTransactions_DistributeRewards(t *testing.T) {
 			validators:             validatorSet,
 			distributeRewardsInput: createTestDistributeRewardsInput(t, 0, validators.GetPublicIdentities(), 10),
 			logger:                 hclog.NewNullLogger(),
+			forks:                  &chain.Forks{chain.Governance: chain.NewFork(0)},
 		}
 
 		distributeRewardsTx, err := fsm.createDistributeRewardsTx()
@@ -621,6 +623,7 @@ func TestFSM_VerifyStateTransactions_DistributeRewards(t *testing.T) {
 			validators:             validatorSet,
 			distributeRewardsInput: createTestDistributeRewardsInput(t, 0, validators.GetPublicIdentities(), 10),
 			logger:                 hclog.NewNullLogger(),
+			forks:                  &chain.Forks{chain.Governance: chain.NewFork(0)},
 		}
 
 		distributeRewardsTx, err := fsm.createDistributeRewardsTx()
@@ -638,6 +641,7 @@ func TestFSM_VerifyStateTransactions_DistributeRewards(t *testing.T) {
 			parent:              &types.Header{Number: 9},
 			validators:          validatorSet,
 			logger:              hclog.NewNullLogger(),
+			forks:               &chain.Forks{chain.Governance: chain.NewFork(0)},
 		}
 
 		err := fsm.VerifyStateTransactions([]*types.Transaction{})