diff --git a/paddle/fluid/platform/device/xpu/tests/enforce_xpu_test.cc b/paddle/fluid/platform/device/xpu/tests/enforce_xpu_test.cc
index b2fd80530feb7462ca34df08acd4651cfd9618c7..afa02f62c324bcccc5012ad85c05a47532ad1f6e 100644
--- a/paddle/fluid/platform/device/xpu/tests/enforce_xpu_test.cc
+++ b/paddle/fluid/platform/device/xpu/tests/enforce_xpu_test.cc
@@ -27,7 +27,7 @@ bool CheckXPUStatusFailure(T value, const std::string& msg) {
   try {
     PADDLE_ENFORCE_XPU_SUCCESS(value);
     return false;
-  } catch (paddle::platform::EnforceNotMet& error) {
+  } catch (phi::enforce::EnforceNotMet& error) {
     std::string ex_msg = error.what();
     std::cout << ex_msg << std::endl;
     return ex_msg.find(msg) != std::string::npos;
@@ -45,7 +45,7 @@ bool CheckXDNNStatusFailure(T value, const std::string& msg) {
   try {
     PADDLE_ENFORCE_XDNN_SUCCESS(value, "XDNN Error ");
     return false;
-  } catch (paddle::platform::EnforceNotMet& error) {
+  } catch (phi::enforce::EnforceNotMet& error) {
     std::string ex_msg = error.what();
     std::cout << ex_msg << std::endl;
     return ex_msg.find(msg) != std::string::npos;