diff --git a/README.md b/README.md
index 8f97dd7..f68cf02 100644
--- a/README.md
+++ b/README.md
@@ -17,7 +17,7 @@
 >
 > ```python
 > import paddle
-> paddle.compat.enable_torch_proxy(scope={"deep_ep"})  # Enable torch proxy before importing deep_ep
+> paddle.enable_compat(scope={"deep_ep"})  # Enable torch proxy before importing deep_ep
 > import deep_ep
 > # use deep_ep
 > ```
diff --git a/tests/test_internode.py b/tests/test_internode.py
index 7b16ddf..4cbd557 100644
--- a/tests/test_internode.py
+++ b/tests/test_internode.py
@@ -1,5 +1,5 @@
 import paddle
-paddle.compat.enable_torch_proxy()
+paddle.enable_compat()
 
 import argparse
 import os
diff --git a/tests/test_intranode.py b/tests/test_intranode.py
index 0d5e29e..2ff610e 100644
--- a/tests/test_intranode.py
+++ b/tests/test_intranode.py
@@ -1,5 +1,5 @@
 import paddle
-paddle.compat.enable_torch_proxy()
+paddle.enable_compat()
 
 import argparse
 import time