diff --git a/python/state-cli.md b/python/state-cli.md
index 1157e01..ac6bb59 100644
--- a/python/state-cli.md
+++ b/python/state-cli.md
@@ -1,7 +1,7 @@
 Implementation of a command line interface that provides different commands
 based on the state of a state machine.
 
-```
+```py
 import prompt_toolkit
 from prompt_toolkit.styles import Style
 from prompt_toolkit.patch_stdout import StdoutProxy, patch_stdout