diff --git a/docs/src/lib/example_discrete_control_space_env.md b/docs/src/lib/example_discrete_control_space_env.md
index 2ff15ec..5b4752f 100644
--- a/docs/src/lib/example_discrete_control_space_env.md
+++ b/docs/src/lib/example_discrete_control_space_env.md
@@ -62,5 +62,7 @@ Now, we can create the environment and the agent and visualize the environment.
 ```@example DiscreteControlSpaceEnv
 env = RxEnvironment(WindyGridWorld((0, 0, 0, 1, 1, 1, 2, 2, 1, 0), [], (8, 5)))
 agent = add!(env, WindyGridWorldAgent((1, 1)))
+```
+```@julia
 RxEnvironments.animate_state(env)
 ```
\ No newline at end of file