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 aaf7c8a99b506fe29141c53ab4d8f19eac563d6a..838bd130aabbad0d400c5dafc41092aaaba3676b 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