diff --git a/debug/accuracy_tools/msprobe/docs/06.data_dump_MindSpore.md b/debug/accuracy_tools/msprobe/docs/06.data_dump_MindSpore.md
index eb4526425a0fd08b2f3bbc69d7d985ec3f294fa3..2582cfc63528a537c7989b4e57494b8ea7c521b8 100644
--- a/debug/accuracy_tools/msprobe/docs/06.data_dump_MindSpore.md
+++ b/debug/accuracy_tools/msprobe/docs/06.data_dump_MindSpore.md
@@ -376,8 +376,11 @@ trainer.train(1, train_dataset, callbacks=[MsprobeStep(debugger)])
 
 ```python
 import mindspore as ms
+from mindspore import set_device
 from mindspore.train import Model
-ms.set_context(mode=ms.PYNATIVE_MODE, device_target="Ascend")
+ms.set_context(mode=ms.PYNATIVE_MODE)
+
+set_device("Ascend", 0)
 
 from msprobe.mindspore import PrecisionDebugger
 from msprobe.mindspore.common.utils import MsprobeStep