AztecProtocol · LHerskind · Mar 19, 2024 · Mar 22, 2024 · Mar 19, 2024 · Mar 22, 2024
diff --git a/avm-transpiler/src/transpile_contract.rs b/avm-transpiler/src/transpile_contract.rs
@@ -16,7 +16,7 @@ pub struct TranspiledContract {
     pub name: String,
     // Functions can be ACIR or AVM
     pub functions: Vec<AvmOrAcirContractFunction>,
-    pub events: serde_json::Value,
+    pub outputs: serde_json::Value,
     pub file_map: serde_json::Value,
     //pub warnings: serde_json::Value,
 }
@@ -29,7 +29,7 @@ pub struct CompiledAcirContract {
     pub noir_version: String,
     pub name: String,
     pub functions: Vec<AcirContractFunction>,
-    pub events: serde_json::Value,
+    pub outputs: serde_json::Value,
     pub file_map: serde_json::Value,
     //pub warnings: serde_json::Value,
 }
@@ -113,7 +113,7 @@ impl From<CompiledAcirContract> for TranspiledContract {
             noir_version: contract.noir_version,
             name: contract.name,
             functions, // some acir, some transpiled avm functions
-            events: contract.events,
+            outputs: contract.outputs,
             file_map: contract.file_map,
             //warnings: contract.warnings,
         }

diff --git a/boxes/boxes/react/src/contracts/src/main.nr b/boxes/boxes/react/src/contracts/src/main.nr
@@ -3,6 +3,7 @@ contract BoxReact {
 
     use dep::value_note::value_note::{ValueNote, VALUE_NOTE_LEN};
 
+    #[aztec(storage)]
     struct Storage {
         numbers: Map<AztecAddress, PrivateMutable<ValueNote>>,
     }

diff --git a/boxes/boxes/vanilla/src/contracts/src/main.nr b/boxes/boxes/vanilla/src/contracts/src/main.nr
@@ -3,6 +3,7 @@ contract Vanilla {
 
     use dep::value_note::value_note::{ValueNote, VALUE_NOTE_LEN};
 
+    #[aztec(storage)]
     struct Storage {
         numbers: Map<AztecAddress, PrivateMutable<ValueNote>>,
     }

diff --git a/docs/docs/developers/contracts/references/storage/main.md b/docs/docs/developers/contracts/references/storage/main.md
@@ -30,11 +30,8 @@ Public state follows the Ethereum style account model, where each contract has i
 
 ## Storage struct
 
-:::info
-The struct **must** be called `Storage` for the Aztec.nr library to properly handle it (this will be relaxed in the future).
-:::
-
 ```rust
+#[aztec(storage)]
 struct Storage {
   // public state variables
   // private state variables

diff --git a/docs/docs/developers/contracts/writing_contracts/storage/define_storage.md b/docs/docs/developers/contracts/writing_contracts/storage/define_storage.md
@@ -8,11 +8,8 @@ To learn more about how storage works in Aztec, read [the concepts](../../../../
 
 [See the storage reference](../../references/storage/main.md).
 
-:::info
-The struct **must** be called `Storage` for the Aztec.nr library to properly handle it (this will be relaxed in the future).
-:::
-
 ```rust
+#[aztec(storage)]
 struct Storage {
   // public state variables
   // private state variables

diff --git a/docs/docs/developers/sandbox/references/cheat_codes.md b/docs/docs/developers/sandbox/references/cheat_codes.md
@@ -461,18 +461,11 @@ The baseSlot is specified in the Aztec.nr contract.
 #### Example
 
 ```rust
+#[aztec(storage)]
 struct Storage {
     balances: Map<AztecAddress, PublicMutable<Field>>,
 }
 
-impl Storage {
-    fn init() -> Self {
-        Storage {
-            balances: Map::new(1, |slot| PublicMutable::new(slot)),
-        }
-    }
-}
-
 contract Token {
     ...
 }
@@ -499,18 +492,11 @@ Note: One Field element occupies a storage slot. Hence, structs with multiple fi
 #### Example
 
 ```rust
+#[aztec(storage)]
 struct Storage {
     balances: Map<AztecAddress, PublicMutable<Field>>,
 }
 
-impl Storage {
-    fn init(context: Context) -> Self {
-        Storage {
-            balances: Map::new(context, 1, |context, slot| PublicMutable::new(context, slot)),
-        }
-    }
-}
-
 contract Token {
     ...
 }
@@ -538,20 +524,12 @@ Note: One Field element occupies a storage slot. Hence, structs with multiple fi
 
 #### Example
 ```rust
+#[aztec(storage)]
 struct Storage {
     ...
     pending_shields: Set<TransparentNote, TRANSPARENT_NOTE_LEN>,
 }
 
-impl Storage {
-    fn init() -> Self {
-        Storage {
-            ...
-            pending_shields: Set::new(context, 5, TransparentNoteMethods),
-        }
-    }
-}
-
 contract Token {
     ...
 }

diff --git a/l1-contracts/src/core/libraries/ConstantsGen.sol b/l1-contracts/src/core/libraries/ConstantsGen.sol
@@ -91,7 +91,7 @@ library Constants {
   uint256 internal constant DEPLOYER_CONTRACT_INSTANCE_DEPLOYED_MAGIC_VALUE =
     0x85864497636cf755ae7bde03f267ce01a520981c21c3682aaf82a631;
   uint256 internal constant DEPLOYER_CONTRACT_ADDRESS =
-    0x1ad693effc2a4a40fdf5911ff29e19d2b6cd3cf73e6c3685519a6fb783144dcb;
+    0x2e89c51c28aefc6b630af43a5908634e4cf552115280eb9fd783bbd224c0bb57;
   uint256 internal constant L1_TO_L2_MESSAGE_ORACLE_CALL_LENGTH = 17;
   uint256 internal constant MAX_NOTE_FIELDS_LENGTH = 20;
   uint256 internal constant GET_NOTE_ORACLE_RETURN_LENGTH = 23;

diff --git a/l1-contracts/test/fixtures/empty_block_0.json b/l1-contracts/test/fixtures/empty_block_0.json
@@ -8,7 +8,7 @@
     "l2ToL1Messages": []
   },
   "block": {
-    "archive": "0x122e56310a16134c202e94b3ba9cb74de123f7ec493e601d2f0b6da00cf57a4a",
+    "archive": "0x28d85d64f43aeffd30919743ecad7e51dbaa60f58dfb98020cbb9ffc29b8b354",
     "body": "0x00000000",
     "txsEffectsHash": "0x00df6b1c97b9e01113fa0363d9ff71c85addc74e92b22d433b2fb082d2493896",
     "decodedHeader": {
@@ -23,8 +23,8 @@
         "chainId": 31337,
         "timestamp": 0,
         "version": 1,
-        "coinbase": "0x66440eb666440eb666440eb666440eb666440eb6",
-        "feeRecipient": "0x061ca689507c7f1ccc68c2ad086c9d5d94f50869cb1b718c6a46aaf77a4500ba"
+        "coinbase": "0xd2335b584d0bbc706a1132378faf166ac32b253b",
+        "feeRecipient": "0x07bc43bbef62d5bb1f694a7ed2d83de7b89ba51b87bbdb5fed59cc92f371cfce"
       },
       "lastArchive": {
         "nextAvailableLeafIndex": 1,
@@ -51,8 +51,8 @@
         }
       }
     },
-    "header": "0x1e3523d3bd50ae6204e1ec2ee1bdf8af4c6217ec80900052d2cf4259379dd13000000001000000000000000000000000000000000000000000000000000000000000000100df6b1c97b9e01113fa0363d9ff71c85addc74e92b22d433b2fb082d249389600089a9d421a82c4a25f7acbebe69e638d5b064fa8a60e018793dcb0be53752c0007638bb56b6dda2b64b8f76841114ac3a87a1820030e2e16772c4d294879c31864fcdaa80ff2719154fa7c8a9050662972707168d69eac9db6fd3110829f800000001016642d9ccd8346c403aa4c3fa451178b22534a27035cdaa6ec34ae53b29c50cb000000800bcfa3e9f1a8922ee92c6dc964d6595907c1804a86753774322b468f69d4f278000001000572c8db882674dd026b8877fbba1b700a4407da3ae9ce5fa43215a28163362b000000800000000000000000000000000000000000000000000000000000000000007a6900000000000000000000000000000000000000000000000000000000000000010000000000000000000000000000000000000000000000000000000000000001000000000000000000000000000000000000000000000000000000000000000066440eb666440eb666440eb666440eb666440eb6061ca689507c7f1ccc68c2ad086c9d5d94f50869cb1b718c6a46aaf77a4500ba",
-    "publicInputsHash": "0x00b833c76dd93572e83879eeb4085a3cd71f24831384fb4d3fe531ed87b859e3",
+    "header": "0x1e3523d3bd50ae6204e1ec2ee1bdf8af4c6217ec80900052d2cf4259379dd13000000001000000000000000000000000000000000000000000000000000000000000000100df6b1c97b9e01113fa0363d9ff71c85addc74e92b22d433b2fb082d249389600089a9d421a82c4a25f7acbebe69e638d5b064fa8a60e018793dcb0be53752c0007638bb56b6dda2b64b8f76841114ac3a87a1820030e2e16772c4d294879c31864fcdaa80ff2719154fa7c8a9050662972707168d69eac9db6fd3110829f800000001016642d9ccd8346c403aa4c3fa451178b22534a27035cdaa6ec34ae53b29c50cb000000800bcfa3e9f1a8922ee92c6dc964d6595907c1804a86753774322b468f69d4f278000001000572c8db882674dd026b8877fbba1b700a4407da3ae9ce5fa43215a28163362b000000800000000000000000000000000000000000000000000000000000000000007a69000000000000000000000000000000000000000000000000000000000000000100000000000000000000000000000000000000000000000000000000000000010000000000000000000000000000000000000000000000000000000000000000d2335b584d0bbc706a1132378faf166ac32b253b07bc43bbef62d5bb1f694a7ed2d83de7b89ba51b87bbdb5fed59cc92f371cfce",
+    "publicInputsHash": "0x0036d6337bcff3e8654b668e271af95bea640bca5d467bcfedf750e329898812",
     "numTxs": 0
   }
 }
diff --git a/l1-contracts/test/fixtures/empty_block_1.json b/l1-contracts/test/fixtures/empty_block_1.json
@@ -8,7 +8,7 @@
     "l2ToL1Messages": []
   },
   "block": {
-    "archive": "0x0d53ab5014f578345d853cd208bda74db64358fef8e5f646240db05893f6e780",
+    "archive": "0x18082b129edbbaabb50b7470846cf28f40b70f75d5709bb8420aa4e7ab2de654",
     "body": "0x00000000",
     "txsEffectsHash": "0x00df6b1c97b9e01113fa0363d9ff71c85addc74e92b22d433b2fb082d2493896",
     "decodedHeader": {
@@ -21,14 +21,14 @@
       "globalVariables": {
         "blockNumber": 2,
         "chainId": 31337,
-        "timestamp": 1711637579,
+        "timestamp": 1712149260,
         "version": 1,
-        "coinbase": "0x66440eb666440eb666440eb666440eb666440eb6",
-        "feeRecipient": "0x061ca689507c7f1ccc68c2ad086c9d5d94f50869cb1b718c6a46aaf77a4500ba"
+        "coinbase": "0xd2335b584d0bbc706a1132378faf166ac32b253b",
+        "feeRecipient": "0x07bc43bbef62d5bb1f694a7ed2d83de7b89ba51b87bbdb5fed59cc92f371cfce"
       },
       "lastArchive": {
         "nextAvailableLeafIndex": 2,
-        "root": "0x122e56310a16134c202e94b3ba9cb74de123f7ec493e601d2f0b6da00cf57a4a"
+        "root": "0x28d85d64f43aeffd30919743ecad7e51dbaa60f58dfb98020cbb9ffc29b8b354"
       },
       "stateReference": {
         "l1ToL2MessageTree": {
@@ -51,8 +51,8 @@
         }
       }
     },
-    "header": "0x122e56310a16134c202e94b3ba9cb74de123f7ec493e601d2f0b6da00cf57a4a00000002000000000000000000000000000000000000000000000000000000000000000100df6b1c97b9e01113fa0363d9ff71c85addc74e92b22d433b2fb082d249389600089a9d421a82c4a25f7acbebe69e638d5b064fa8a60e018793dcb0be53752c0007638bb56b6dda2b64b8f76841114ac3a87a1820030e2e16772c4d294879c31864fcdaa80ff2719154fa7c8a9050662972707168d69eac9db6fd3110829f800000002016642d9ccd8346c403aa4c3fa451178b22534a27035cdaa6ec34ae53b29c50cb000001000bcfa3e9f1a8922ee92c6dc964d6595907c1804a86753774322b468f69d4f278000001800572c8db882674dd026b8877fbba1b700a4407da3ae9ce5fa43215a28163362b000000c00000000000000000000000000000000000000000000000000000000000007a6900000000000000000000000000000000000000000000000000000000000000010000000000000000000000000000000000000000000000000000000000000002000000000000000000000000000000000000000000000000000000006605844b66440eb666440eb666440eb666440eb666440eb6061ca689507c7f1ccc68c2ad086c9d5d94f50869cb1b718c6a46aaf77a4500ba",
-    "publicInputsHash": "0x00035031752fa374209f1ec4bc08613cae25ba65779c10130681421f6e71c6d9",
+    "header": "0x28d85d64f43aeffd30919743ecad7e51dbaa60f58dfb98020cbb9ffc29b8b35400000002000000000000000000000000000000000000000000000000000000000000000100df6b1c97b9e01113fa0363d9ff71c85addc74e92b22d433b2fb082d249389600089a9d421a82c4a25f7acbebe69e638d5b064fa8a60e018793dcb0be53752c0007638bb56b6dda2b64b8f76841114ac3a87a1820030e2e16772c4d294879c31864fcdaa80ff2719154fa7c8a9050662972707168d69eac9db6fd3110829f800000002016642d9ccd8346c403aa4c3fa451178b22534a27035cdaa6ec34ae53b29c50cb000001000bcfa3e9f1a8922ee92c6dc964d6595907c1804a86753774322b468f69d4f278000001800572c8db882674dd026b8877fbba1b700a4407da3ae9ce5fa43215a28163362b000000c00000000000000000000000000000000000000000000000000000000000007a690000000000000000000000000000000000000000000000000000000000000001000000000000000000000000000000000000000000000000000000000000000200000000000000000000000000000000000000000000000000000000660d530cd2335b584d0bbc706a1132378faf166ac32b253b07bc43bbef62d5bb1f694a7ed2d83de7b89ba51b87bbdb5fed59cc92f371cfce",
+    "publicInputsHash": "0x00056a5ebf2e50d61c9dab6ff88579d680c284f8ad94a2c7e249fb3cc65be3fb",
     "numTxs": 0
   }
 }
diff --git a/l1-contracts/test/fixtures/mixed_block_0.json b/l1-contracts/test/fixtures/mixed_block_0.json
diff --git a/l1-contracts/test/fixtures/mixed_block_1.json b/l1-contracts/test/fixtures/mixed_block_1.json
diff --git a/noir-projects/aztec-nr/aztec/src/prelude.nr b/noir-projects/aztec-nr/aztec/src/prelude.nr
@@ -4,7 +4,7 @@ use crate::{
     state_vars::{
     map::Map, private_immutable::PrivateImmutable, private_mutable::PrivateMutable,
     public_immutable::PublicImmutable, public_mutable::PublicMutable, private_set::PrivateSet,
-    shared_immutable::SharedImmutable
+    shared_immutable::SharedImmutable, storage::Storable
 },
     log::{emit_unencrypted_log, emit_encrypted_log}, context::PrivateContext,
     note::{

diff --git a/noir-projects/aztec-nr/aztec/src/state_vars/storage.nr b/noir-projects/aztec-nr/aztec/src/state_vars/storage.nr
@@ -5,3 +5,9 @@ trait Storage<T> where T: Serialize<N> + Deserialize<N> {
         self.storage_slot
     }
 }
+
+struct Storable<N> {
+        slot: Field,
+        typ: str<N>
+    }
+
diff --git a/noir-projects/noir-contracts/contracts/app_subscription_contract/src/main.nr b/noir-projects/noir-contracts/contracts/app_subscription_contract/src/main.nr
@@ -17,6 +17,7 @@ contract AppSubscription {
 
     use crate::subscription_note::{SubscriptionNote, SUBSCRIPTION_NOTE_LEN};
 
+    #[aztec(storage)]
     struct Storage {
         // The following is only needed in private but we use ShareImmutable here instead of PrivateImmutable because
         // the value can be publicly known and SharedImmutable provides us with a better devex here because we don't

diff --git a/noir-projects/noir-contracts/contracts/avm_test_contract/src/main.nr b/noir-projects/noir-contracts/contracts/avm_test_contract/src/main.nr
@@ -35,6 +35,7 @@ contract AvmTest {
     // avm lib
     use dep::aztec::avm::hash::{keccak256, poseidon, sha256};
 
+    #[aztec(storage)]
     struct Storage {
         single: PublicMutable<Field>,
         list: PublicMutable<Note>,

diff --git a/noir-projects/noir-contracts/contracts/benchmarking_contract/src/main.nr b/noir-projects/noir-contracts/contracts/benchmarking_contract/src/main.nr
@@ -12,6 +12,7 @@ contract Benchmarking {
 
     use dep::aztec::{context::Context};
 
+    #[aztec(storage)]
     struct Storage {
         notes: Map<AztecAddress, PrivateSet<ValueNote>>,
         balances: Map<AztecAddress, PublicMutable<Field>>,

diff --git a/noir-projects/noir-contracts/contracts/card_game_contract/src/main.nr b/noir-projects/noir-contracts/contracts/card_game_contract/src/main.nr
@@ -12,6 +12,7 @@ contract CardGame {
     use crate::cards::{PACK_CARDS, Deck, Card, get_pack_cards, compute_deck_strength};
     use crate::game::{NUMBER_OF_PLAYERS, NUMBER_OF_CARDS_DECK, PLAYABLE_CARDS, PlayerEntry, Game, GAME_SERIALIZED_LEN};
 
+    #[aztec(storage)]
     struct Storage {
         collections: Map<AztecAddress, Deck>,
         game_decks: Map<Field, Map<AztecAddress, Deck>>,

diff --git a/noir-projects/noir-contracts/contracts/child_contract/src/main.nr b/noir-projects/noir-contracts/contracts/child_contract/src/main.nr
@@ -8,6 +8,7 @@ contract Child {
     };
     use dep::value_note::value_note::ValueNote;
 
+    #[aztec(storage)]
     struct Storage {
         current_value: PublicMutable<Field>,
         a_private_value: PrivateSet<ValueNote>,

diff --git a/noir-projects/noir-contracts/contracts/claim_contract/src/main.nr b/noir-projects/noir-contracts/contracts/claim_contract/src/main.nr
@@ -9,6 +9,7 @@ contract Claim {
     use dep::value_note::value_note::ValueNote;
     use interfaces::Token;
 
+    #[aztec(storage)]
     struct Storage {
       // Address of a contract based on whose notes we distribute the rewards
       target_contract: SharedImmutable<AztecAddress>,

diff --git a/...oir-contracts/contracts/contract_class_registerer_contract/src/events/class_registered.nr b/...oir-contracts/contracts/contract_class_registerer_contract/src/events/class_registered.nr
@@ -4,7 +4,7 @@ use dep::aztec::protocol_types::{
     traits::Serialize
 };
 
-// #[event]
+// #[aztec(event)]
 struct ContractClassRegistered {
     contract_class_id: ContractClassId,
     version: Field,

diff --git a/...s/contracts/contract_class_registerer_contract/src/events/private_function_broadcasted.nr b/...s/contracts/contract_class_registerer_contract/src/events/private_function_broadcasted.nr
@@ -31,7 +31,7 @@ impl Serialize<MAX_PACKED_BYTECODE_SIZE_PER_PRIVATE_FUNCTION_IN_FIELDS + 3> for
     }
 }
 
-// #[event]
+// #[aztec(event)]
 struct ClassPrivateFunctionBroadcasted {
     contract_class_id: ContractClassId,
     artifact_metadata_hash: Field,

diff --git a/...racts/contract_class_registerer_contract/src/events/unconstrained_function_broadcasted.nr b/...racts/contract_class_registerer_contract/src/events/unconstrained_function_broadcasted.nr
@@ -28,7 +28,7 @@ impl Serialize<MAX_PACKED_BYTECODE_SIZE_PER_UNCONSTRAINED_FUNCTION_IN_FIELDS + 2
     }
 }
 
-// #[event]
+// #[aztec(event)]
 struct ClassUnconstrainedFunctionBroadcasted {
     contract_class_id: ContractClassId,
     artifact_metadata_hash: Field,

diff --git a/...r-contracts/contracts/contract_instance_deployer_contract/src/events/instance_deployed.nr b/...r-contracts/contracts/contract_instance_deployer_contract/src/events/instance_deployed.nr
@@ -4,7 +4,7 @@ use dep::aztec::protocol_types::{
     constants::DEPLOYER_CONTRACT_INSTANCE_DEPLOYED_MAGIC_VALUE, traits::Serialize
 };
 
-// #[event]
+// #[aztec(event)]
 struct ContractInstanceDeployed {
     address: AztecAddress,
     version: u8,

diff --git a/noir-projects/noir-contracts/contracts/counter_contract/src/main.nr b/noir-projects/noir-contracts/contracts/counter_contract/src/main.nr
@@ -7,6 +7,7 @@ contract Counter {
     // docs:end:imports
 
     // docs:start:storage_struct
+    #[aztec(storage)]
     struct Storage {
         counters: Map<AztecAddress, EasyPrivateUint>,
     }

diff --git a/noir-projects/noir-contracts/contracts/crowdfunding_contract/src/main.nr b/noir-projects/noir-contracts/contracts/crowdfunding_contract/src/main.nr
@@ -9,7 +9,7 @@ contract Crowdfunding {
     use dep::value_note::value_note::ValueNote;
     use interfaces::Token;
 
-    #[event]
+    #[aztec(event)]
     struct WithdrawalProcessed {
         who: AztecAddress,
         amount: u64,
@@ -21,6 +21,7 @@ contract Crowdfunding {
         }
     }
 
+    #[aztec(storage)]
     struct Storage {
         // Token used for donations (e.g. DAI)
         donation_token: SharedImmutable<AztecAddress>,

diff --git a/noir-projects/noir-contracts/contracts/delegated_on_contract/src/main.nr b/noir-projects/noir-contracts/contracts/delegated_on_contract/src/main.nr
@@ -6,6 +6,7 @@ contract DelegatedOn {
     };
     use dep::value_note::value_note::ValueNote;
 
+    #[aztec(storage)]
     struct Storage {
         current_value: PublicMutable<Field>,
         a_private_value: PrivateSet<ValueNote>,

diff --git a/noir-projects/noir-contracts/contracts/delegator_contract/src/main.nr b/noir-projects/noir-contracts/contracts/delegator_contract/src/main.nr
@@ -6,6 +6,7 @@ contract Delegator {
     };
     use dep::value_note::value_note::ValueNote;
 
+    #[aztec(storage)]
     struct Storage {
         current_value: PublicMutable<Field>,
         a_private_value: PrivateSet<ValueNote>,

diff --git a/noir-projects/noir-contracts/contracts/docs_example_contract/src/main.nr b/noir-projects/noir-contracts/contracts/docs_example_contract/src/main.nr
@@ -23,6 +23,7 @@ contract DocsExample {
     use crate::options::create_account_card_getter_options;
     use crate::types::{card_note::{CardNote, CARD_NOTE_LEN}, leader::Leader};
 
+    #[aztec(storage)]
     struct Storage {
         // Shows how to create a custom struct in PublicMutable
         // docs:start:storage-leader-declaration
@@ -118,7 +119,7 @@ contract DocsExample {
             context.this_address(),
             FunctionSelector::from_signature("get_shared_immutable_constrained_private()")
         );
-        Leader::deserialize([ret[0], ret[1]])
+        Leader::deserialize([ret[0], ret[1] + 1])
     }
 
     #[aztec(public)]
@@ -131,7 +132,7 @@ contract DocsExample {
             context.this_address(),
             FunctionSelector::from_signature("get_shared_immutable_constrained_public()")
         );
-        Leader::deserialize([ret[0], ret[1]])
+        Leader::deserialize([ret[0], ret[1] + 1])
     }
 
     #[aztec(public)]

diff --git a/noir-projects/noir-contracts/contracts/easy_private_token_contract/src/main.nr b/noir-projects/noir-contracts/contracts/easy_private_token_contract/src/main.nr
@@ -4,6 +4,7 @@ contract EasyPrivateToken {
     use dep::value_note::{balance_utils, value_note::ValueNote};
     use dep::easy_private_state::EasyPrivateUint;
 
+    #[aztec(storage)]
     struct Storage {
         balances: Map<AztecAddress, EasyPrivateUint>,
     }

diff --git a/noir-projects/noir-contracts/contracts/easy_private_voting_contract/src/main.nr b/noir-projects/noir-contracts/contracts/easy_private_voting_contract/src/main.nr
@@ -7,6 +7,7 @@ contract EasyPrivateVoting {
     use dep::aztec::context::Context;
     // docs:end:imports
     // docs:start:storage_struct
+    #[aztec(storage)]
     struct Storage {
         admin: PublicMutable<AztecAddress>, // admin can end vote
         tally: Map<Field, PublicMutable<Field>>, // we will store candidate as key and number of votes as value