diff --git a/docs/tutorials/accelerate-pytorch/pytorch.md b/docs/tutorials/accelerate-pytorch/pytorch.md
index 69474224ed..dbceb98b2c 100644
--- a/docs/tutorials/accelerate-pytorch/pytorch.md
+++ b/docs/tutorials/accelerate-pytorch/pytorch.md
@@ -28,7 +28,7 @@ At the heart of PyTorch is the `nn.Module`, a class that represents an entire de
 
 To code an image processing model using PyTorch itself:
 
-```bash
+```python
 import torch
 import torch.nn as nn
 import torchvision.transforms as T