diff --git a/esm/README.md b/esm/README.md
index 5cda12cc..b2c9d24c 100644
--- a/esm/README.md
+++ b/esm/README.md
@@ -56,7 +56,7 @@ logits = result["logits"]  # Shape: (batch, length, vocab_size)
 
 ### Masked Language Modeling
 
-```python
+```bash
 # For a complete example, see main.py
 python main.py --sequence "YOUR_SEQUENCE" --mask-position 50
 ```