diff --git a/README.md b/README.md
index ab21aec783..b99a8cc955 100644
--- a/README.md
+++ b/README.md
@@ -35,7 +35,7 @@ Kernel Library for LLM Serving ❤️ PaddlePadddle
 >
 > ```python
 > import paddle
-> paddle.compat.enable_torch_proxy(scope={"flashinfer"})  # Enable torch proxy before importing flashinfer
+> paddle.enable_compat(scope={"flashinfer"})  # Enable torch proxy before importing flashinfer
 > import flashinfer
 > # use flashinfer
 > ```
diff --git a/tests/attention/test_attention_sink_blackwell.py b/tests/attention/test_attention_sink_blackwell.py
index 19c379832c..867c990f45 100644
--- a/tests/attention/test_attention_sink_blackwell.py
+++ b/tests/attention/test_attention_sink_blackwell.py
@@ -16,7 +16,7 @@
 
 import paddle
 
-paddle.compat.enable_torch_proxy()
+paddle.enable_compat()
 import einops
 import pytest
 import torch
diff --git a/tests/comm/test_trtllm_allreduce_fusion_paddle.py b/tests/comm/test_trtllm_allreduce_fusion_paddle.py
index f5b61ca9eb..50528ad13a 100644
--- a/tests/comm/test_trtllm_allreduce_fusion_paddle.py
+++ b/tests/comm/test_trtllm_allreduce_fusion_paddle.py
@@ -73,7 +73,7 @@ def kernel(workspace_tensor, rank, world_size):
 
 def _run_simple_worker(world_size, rank, distributed_init_port):
     # Create workspace
-    # paddle.compat.enable_torch_proxy()
+    # paddle.enable_compat()
     # Set all required environment variables
     os.environ["FLAGS_SELECTED_GPUS"] = str(rank)  # Key: set GPU ID
     os.environ["PADDLE_TRAINER_ID"] = str(rank)
diff --git a/tests/conftest.py b/tests/conftest.py
index 58c44ab6b8..f23f0d6290 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -6,7 +6,7 @@
 
 import paddle
 
-paddle.compat.enable_torch_proxy()
+paddle.enable_compat()
 import pytest
 import torch
 # from torch.torch_version import TorchVersion
diff --git a/tests/moe/test_trtllm_gen_fused_moe.py b/tests/moe/test_trtllm_gen_fused_moe.py
index 1543895b07..cbac2dff62 100644
--- a/tests/moe/test_trtllm_gen_fused_moe.py
+++ b/tests/moe/test_trtllm_gen_fused_moe.py
@@ -16,7 +16,7 @@
 
 import paddle
 
-paddle.compat.enable_torch_proxy()
+paddle.enable_compat()
 import functools
 from typing import Tuple
 from abc import ABC, abstractmethod