diff --git a/test/src/test/java/org/apache/seata/core/rpc/netty/mockserver/MockServerTest.java b/test/src/test/java/org/apache/seata/core/rpc/netty/mockserver/MockServerTest.java
index 887c7274f36..2e8c568f9bc 100644
--- a/test/src/test/java/org/apache/seata/core/rpc/netty/mockserver/MockServerTest.java
+++ b/test/src/test/java/org/apache/seata/core/rpc/netty/mockserver/MockServerTest.java
@@ -83,7 +83,7 @@ private static String doTestCommit(int times) throws TransactionException {
 
         String xid = tm.begin(ProtocolTestConstants.APPLICATION_ID, ProtocolTestConstants.SERVICE_GROUP, "test", 60000);
         MockCoordinator.getInstance().setExpectedRetry(xid, times);
-        Long branchId = rm.branchRegister(BranchType.AT, RESOURCE_ID, "1", xid, "1", "1");
+        Long branchId = rm.branchRegister(BranchType.TCC, RESOURCE_ID, "1", xid, "1", "1");
         GlobalStatus commit = tm.commit(xid);
         Assertions.assertEquals(GlobalStatus.Committed, commit);
         return xid;
@@ -96,7 +96,7 @@ private static String doTestRollback(int times) throws TransactionException {
 
         String xid = tm.begin(ProtocolTestConstants.APPLICATION_ID, ProtocolTestConstants.SERVICE_GROUP, "test", 60000);
         MockCoordinator.getInstance().setExpectedRetry(xid, times);
-        Long branchId = rm.branchRegister(BranchType.AT, RESOURCE_ID, "1", xid, "1", "1");
+        Long branchId = rm.branchRegister(BranchType.TCC, RESOURCE_ID, "1", xid, "1", "1");
         GlobalStatus rollback = tm.rollback(xid);
         Assertions.assertEquals(GlobalStatus.Rollbacked, rollback);
         return xid;