diff --git a/bench/ssz/encode_decode_bench.exs b/bench/ssz/encode_decode_bench.exs
index 1c0e79c7c..4b7a6e309 100644
--- a/bench/ssz/encode_decode_bench.exs
+++ b/bench/ssz/encode_decode_bench.exs
@@ -1,4 +1,3 @@
-alias LambdaEthereumConsensus.SszEx
 alias LambdaEthereumConsensus.Store.StateDb
 alias Types.BeaconState
 alias Types.Checkpoint
@@ -6,7 +5,7 @@ alias Types.Checkpoint
 # To run these benchmarks, you need a BeaconState stored in the Db beforehand.
 # The --mode db flag is also needed.
 
-{:ok, state} = StateDb.get_latest_state()
+{:ok, %{beacon_state: state}} = StateDb.get_latest_state()
 :mainnet = ChainSpec.get_preset()
 {:ok, encoded_state} = Ssz.to_ssz(state)
 
diff --git a/bench/ssz/hash_tree_root_bench.exs b/bench/ssz/hash_tree_root_bench.exs
index 320404b07..697d1697c 100644
--- a/bench/ssz/hash_tree_root_bench.exs
+++ b/bench/ssz/hash_tree_root_bench.exs
@@ -1,12 +1,11 @@
-alias LambdaEthereumConsensus.SszEx
-alias LambdaEthereumConsensus.SszEx.Merkleization
 alias LambdaEthereumConsensus.Store.StateDb
+alias SszEx.Merkleization
 alias Types.BeaconState
 
 # To run these benchmarks, you need a BeaconState stored in the Db beforehand.
 # The --mode db flag is also needed.
 
-{:ok, state} = StateDb.get_latest_state()
+{:ok, %{beacon_state: state}} = StateDb.get_latest_state()
 :mainnet = ChainSpec.get_preset()
 
 Benchee.run(