fix: update tree height

Author: ypatil12

beacon/beacon_state_top_level_roots.go

@@ -99,11 +99,14 @@ type BeaconStateTopLevelRootsElectra struct {
 
 func ProveBeaconTopLevelRootAgainstBeaconState(beaconTopLevelRoots *VersionedBeaconStateTopLevelRoots, index uint64) (common.Proof, error) {
 	var v reflect.Value
+	var treeHeight uint64
 	switch beaconTopLevelRoots.Version {
 	case spec.DataVersionDeneb:
 		v = reflect.ValueOf(*beaconTopLevelRoots.Deneb)
+		treeHeight = BEACON_STATE_TREE_HEIGHT_DENEB
 	case spec.DataVersionElectra:
 		v = reflect.ValueOf(*beaconTopLevelRoots.Electra)
+		treeHeight = BEACON_STATE_TREE_HEIGHT_ELECTRA
 	default:
 		return nil, errors.New("unsupported beacon state version")
 	}
@@ -120,5 +123,5 @@ func ProveBeaconTopLevelRootAgainstBeaconState(beaconTopLevelRoots *VersionedBea
 		roots[i] = v.(phase0.Root)
 	}
 
-	return common.GetProof(roots, index, BEACON_STATE_TREE_HEIGHT_DENEB)
+	return common.GetProof(roots, index, treeHeight)
 }

beacon/constants.go

@@ -3,11 +3,11 @@ package beacon
 const (
 	BEACON_BLOCK_HEADER_NUM_FIELDS = uint64(5)
 
-	BEACON_BLOCK_HEADER_TREE_HEIGHT = uint64(3)
-	BEACON_STATE_TREE_HEIGHT_DENEB  = uint64(5)
-	BEACON_STATE_TREE_HEIGHT_PECTRA = uint64(6)
-	BALANCE_TREE_HEIGHT             = uint64(38)
-	VALIDATOR_TREE_HEIGHT           = uint64(40)
+	BEACON_BLOCK_HEADER_TREE_HEIGHT  = uint64(3)
+	BEACON_STATE_TREE_HEIGHT_DENEB   = uint64(5)
+	BEACON_STATE_TREE_HEIGHT_ELECTRA = uint64(6)
+	BALANCE_TREE_HEIGHT              = uint64(38)
+	VALIDATOR_TREE_HEIGHT            = uint64(40)
 
 	STATE_ROOT_INDEX = uint64(3)