diff --git a/docs/source/en/features/distributed_optimizers.md b/docs/source/en/features/distributed_optimizers.md
index eab8d770834a..7590669dfd2d 100644
--- a/docs/source/en/features/distributed_optimizers.md
+++ b/docs/source/en/features/distributed_optimizers.md
@@ -138,4 +138,4 @@ optim = DistGaloreAwamW(
   </tr>
 </table>
 
-<!-- doc-test-command: python test_dist_optim.py  -->
+<!-- doc-test-command: colossalai run --nproc_per_node 4 distributed_optimizers.py  -->
diff --git a/docs/source/zh-Hans/features/distributed_optimizers.md b/docs/source/zh-Hans/features/distributed_optimizers.md
index 434a3d926a5f..e1d00d22ace1 100644
--- a/docs/source/zh-Hans/features/distributed_optimizers.md
+++ b/docs/source/zh-Hans/features/distributed_optimizers.md
@@ -138,4 +138,4 @@ optim = DistGaloreAwamW(
   </tr>
 </table>
 
-<!-- doc-test-command: python test_dist_optim.py  -->
+<!-- doc-test-command: colossalai run --nproc_per_node 4 distributed_optimizers.py  -->