diff --git a/docs/docs/developers/guides/js_apps/index.md b/docs/docs/developers/guides/js_apps/index.md
index f3835d3a50b6..2e6512205a16 100644
--- a/docs/docs/developers/guides/js_apps/index.md
+++ b/docs/docs/developers/guides/js_apps/index.md
@@ -17,7 +17,7 @@ npm install @aztec/aztec.js
 
 At the top of your JavaScript file, you can import what you need, eg:
 
-#include_code import_aztecjs yarn-project/end-to-end/src/e2e_simple.test.ts typescript
+#include_code import_aztecjs yarn-project/end-to-end/src/e2e_2_pxes.test.ts typescript
 
 ## Flow
 
@@ -26,4 +26,4 @@ These are some of the important functions you'll need to use in your Aztec.js:
 - [Create an account with `@aztec/accounts`](./create_account.md)
 - [Deploy a contract](./deploy_contract.md)
 - [Simulate a function call](./call_view_function.md)
-- [Send a transaction](./send_transaction.md)
\ No newline at end of file
+- [Send a transaction](./send_transaction.md)
diff --git a/yarn-project/end-to-end/src/e2e_2_pxes.test.ts b/yarn-project/end-to-end/src/e2e_2_pxes.test.ts
index 79de4f212ff6..39fdb0c3effb 100644
--- a/yarn-project/end-to-end/src/e2e_2_pxes.test.ts
+++ b/yarn-project/end-to-end/src/e2e_2_pxes.test.ts
@@ -1,6 +1,8 @@
 import { getSchnorrAccount } from '@aztec/accounts/schnorr';
 import { type InitialAccountData, deployFundedSchnorrAccount } from '@aztec/accounts/testing';
+// docs:start:import_aztecjs
 import { type AztecAddress, type AztecNode, Fr, type Logger, type PXE, type Wallet, sleep } from '@aztec/aztec.js';
+// docs:end:import_aztecjs
 import { ChildContract } from '@aztec/noir-contracts.js/Child';
 import { TokenContract } from '@aztec/noir-contracts.js/Token';
 
diff --git a/yarn-project/end-to-end/src/e2e_simple.test.ts b/yarn-project/end-to-end/src/e2e_simple.test.ts
index 6ac62b282c98..cee7e2414ab0 100644
--- a/yarn-project/end-to-end/src/e2e_simple.test.ts
+++ b/yarn-project/end-to-end/src/e2e_simple.test.ts
@@ -1,7 +1,5 @@
-// docs:start:import_aztecjs
 import type { AztecNodeConfig } from '@aztec/aztec-node';
 import { type AztecNode, ContractDeployer, Fr, type Wallet, waitForProven } from '@aztec/aztec.js';
-// docs:end:import_aztecjs
 // eslint-disable-next-line no-restricted-imports
 import { EthAddress } from '@aztec/foundation/eth-address';
 import { StatefulTestContractArtifact } from '@aztec/noir-contracts.js/StatefulTest';