diff --git a/docs/docs/tutorials/codealong/contract_tutorials/counter_contract.md b/docs/docs/tutorials/codealong/contract_tutorials/counter_contract.md
index 017643618aa7..3cada7c837f0 100644
--- a/docs/docs/tutorials/codealong/contract_tutorials/counter_contract.md
+++ b/docs/docs/tutorials/codealong/contract_tutorials/counter_contract.md
@@ -61,7 +61,7 @@ easy_private_state = { git="https://github.com/AztecProtocol/aztec-packages/", t
 Go to `main.nr` and start with this contract initialization:
 
 ```rust
-contract Counter {
+#include_code setup /noir-projects/noir-contracts/contracts/counter_contract/src/main.nr raw
 }
 ```
 
@@ -159,4 +159,4 @@ Follow the private voting contract tutorial on the [next page](./private_voting_
 
 ### Optional: Learn more about concepts mentioned here
 
- - [Functions and annotations like `#[aztec(private)]`](../../../aztec/smart_contracts/functions/inner_workings.md)
+- [Functions and annotations like `#[aztec(private)]`](../../../aztec/smart_contracts/functions/inner_workings.md)
diff --git a/noir-projects/noir-contracts/contracts/counter_contract/src/main.nr b/noir-projects/noir-contracts/contracts/counter_contract/src/main.nr
index 42bc8c396f8f..ed625c82a5ad 100644
--- a/noir-projects/noir-contracts/contracts/counter_contract/src/main.nr
+++ b/noir-projects/noir-contracts/contracts/counter_contract/src/main.nr
@@ -1,7 +1,9 @@
+// docs:start:setup
 use dep::aztec::macros::aztec;
 
 #[aztec]
 contract Counter {
+    // docs:end:setup
     // docs:start:imports
     use aztec::prelude::{AztecAddress, Map};
     use value_note::{balance_utils, value_note::ValueNote};