diff --git a/src/ethereum_optimized/state_db.py b/src/ethereum_optimized/state_db.py
index fd1f48f2c9..32566723a6 100644
--- a/src/ethereum_optimized/state_db.py
+++ b/src/ethereum_optimized/state_db.py
@@ -345,7 +345,7 @@ def get_storage(state: State, address: Address, key: Bytes32) -> U256:
 
     @add_item(patches)
     def get_storage_original(
-        state: State, address: Address, key: Bytes
+        state: State, address: Address, key: Bytes32
     ) -> U256:
         """
         See `state`.