diff --git a/docs/guides/use-environment-variables.md b/docs/guides/use-environment-variables.md
index 086567e68..3da6fa226 100644
--- a/docs/guides/use-environment-variables.md
+++ b/docs/guides/use-environment-variables.md
@@ -466,6 +466,8 @@ export function registerShell(env: Environment) {
 }
 ```
 
+### Register variables
+
 Then, update the `registerShell` function to register the `apiBaseUrl` environment variable based on the provided `env` argument:
 
 ```ts !#41-44 shell/src/register.tsx
@@ -517,6 +519,8 @@ export function registerShell(env: Environment) {
 }
 ```
 
+### Module augmentation
+
 Then, augment the `EnvironmentVariables` TypeScript interface to include the `apiBaseUrl` variable:
 
 ```ts !#6 shell/types/env-vars.d.ts