diff --git a/lib/babe/helpers_test.go b/lib/babe/helpers_test.go
index 7063027e8b..3479b7ddf7 100644
--- a/lib/babe/helpers_test.go
+++ b/lib/babe/helpers_test.go
@@ -199,7 +199,8 @@ func createTestService(t *testing.T, cfg ServiceConfig, genesis genesis.Genesis,
 
 	// Allow for epoch state to be made from custom babe config
 	if babeConfig != nil {
-		dbSrv.Epoch, err = state.NewEpochStateFromGenesis(dbSrv.DB(), dbSrv.Block, babeConfig)
+		dbSrv.Epoch, err = state.NewEpochStateFromGenesis(dbSrv.DB(),
+			dbSrv.Base, dbSrv.Block, babeConfig)
 		require.NoError(t, err)
 	}
 	cfg.BlockState = dbSrv.Block