diff --git a/torch_xla/csrc/runtime/xla_util_test.cc b/torch_xla/csrc/runtime/xla_util_test.cc
index 6119ef73092..7b45f7daab9 100644
--- a/torch_xla/csrc/runtime/xla_util_test.cc
+++ b/torch_xla/csrc/runtime/xla_util_test.cc
@@ -12,10 +12,10 @@
 #include "tsl/platform/errors.h"
 #include "tsl/platform/protobuf.h"
 #include "tsl/platform/status_matchers.h"
-#include "tsl/protobuf/error_codes.pb.h"
 #include "xla/client/xla_builder.h"
 #include "xla/client/xla_computation.h"
 #include "xla/tsl/lib/core/status_test_util.h"
+#include "xla/tsl/protobuf/error_codes.pb.h"
 #include "xla_util.h"
 
 namespace torch_xla {