diff --git a/examples/images/diffusion/README.md b/examples/images/diffusion/README.md
index 02da1e536..eb899c563 100644
--- a/examples/images/diffusion/README.md
+++ b/examples/images/diffusion/README.md
@@ -46,6 +46,7 @@ pip install -e .
 
 ```
 git clone https://github.com/1SAA/lightning.git
+cd lightning
 git checkout strategy/colossalai
 export PACKAGE_NAME=pytorch
 pip install .