fixed return type issues

Layr-Labs · Feb 28, 2024 · d9c7071 · d9c7071
1 parent 2c67b10
commit d9c7071
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/eigen_pod_proofs.go b/eigen_pod_proofs.go
@@ -1,7 +1,6 @@
 package eigenpodproofs
 
 import (
-	"encoding/json"
 	"errors"
 	"time"
 
@@ -54,7 +53,7 @@ func (epp *EigenPodProofs) ComputeBeaconStateRoot(beaconState *deneb.BeaconState
 			if err != nil {
 				return nil, err
 			}
-			return stateRoot[:], nil
+			return stateRoot, nil
 		},
 	)
 	if err != nil {
@@ -84,7 +83,7 @@ func (epp *EigenPodProofs) ComputeBeaconStateTopLevelRoots(beaconState *spec.Ver
 			if err != nil {
 				return nil, err
 			}
-			return json.Marshal(beaconStateTopLevelRoots)
+			return beaconStateTopLevelRoots, nil
 		},
 	)
 	if err != nil {
@@ -93,7 +92,7 @@ func (epp *EigenPodProofs) ComputeBeaconStateTopLevelRoots(beaconState *spec.Ver
 
 	beaconStateTopLevelRoots, ok := beaconStateTopLevelRootsInterface.(*beacon.BeaconStateTopLevelRoots)
 	if !ok {
-		return nil, errors.New("beacon state top level roots is not of type []byte")
+		return nil, errors.New("beacon state top level roots is not of type *beacon.BeaconStateTopLevelRoots")
 	}
 	return beaconStateTopLevelRoots, err
 }