diff --git a/alpha/alpha-server/pom.xml b/alpha/alpha-server/pom.xml
index 857068edf..e1d6a8c5e 100644
--- a/alpha/alpha-server/pom.xml
+++ b/alpha/alpha-server/pom.xml
@@ -170,7 +170,7 @@
     <profile>
       <id>spring-boot-2</id>
       <properties>
-        <spring.boot.version>2.0.0.RELEASE</spring.boot.version>
+        <spring.boot.version>${spring.boot2.version}</spring.boot.version>
       </properties>
       <!-- Need to add the dependency of log4j2 core -->
       <dependencies>
diff --git a/omega/omega-spring-tx/pom.xml b/omega/omega-spring-tx/pom.xml
index 8cfca6743..9be47748b 100644
--- a/omega/omega-spring-tx/pom.xml
+++ b/omega/omega-spring-tx/pom.xml
@@ -112,4 +112,13 @@
 
   </dependencies>
 
+  <profiles>
+    <profile>
+      <id>spring-boot-2</id>
+      <properties>
+        <spring.boot.version>${spring.boot2.version}</spring.boot.version>
+      </properties>
+    </profile>
+  </profiles>
+
 </project>
diff --git a/omega/omega-spring-tx/src/test/java/org/apache/servicecomb/saga/omega/transaction/spring/TransactionInterceptionTest.java b/omega/omega-spring-tx/src/test/java/org/apache/servicecomb/saga/omega/transaction/spring/TransactionInterceptionTest.java
index 9fd2a7ed9..2ec42a512 100644
--- a/omega/omega-spring-tx/src/test/java/org/apache/servicecomb/saga/omega/transaction/spring/TransactionInterceptionTest.java
+++ b/omega/omega-spring-tx/src/test/java/org/apache/servicecomb/saga/omega/transaction/spring/TransactionInterceptionTest.java
@@ -136,7 +136,7 @@ public void sendsUserToRemote_AroundTransaction() throws Exception {
         toArray(messages)
     );
 
-    User actual = userRepository.findOne(user.id());
+    User actual = userRepository.findByUsername(user.username());
     assertThat(actual, is(user));
   }
 
@@ -169,8 +169,8 @@ public void compensateOnTransactionException() throws Exception {
     messageHandler.onReceive(globalTxId, newLocalTxId, globalTxId, compensationMethod, user);
     messageHandler.onReceive(globalTxId, anotherLocalTxId, localTxId, compensationMethod, anotherUser);
 
-    assertThat(userRepository.findOne(user.id()), is(nullValue()));
-    assertThat(userRepository.findOne(anotherUser.id()), is(nullValue()));
+    assertThat(userRepository.findByUsername(user.username()), is(nullValue()));
+    assertThat(userRepository.findByUsername(anotherUser.username()), is(nullValue()));
 
     assertArrayEquals(
         new String[]{
diff --git a/omega/omega-spring-tx/src/test/java/org/apache/servicecomb/saga/omega/transaction/spring/TransactionalUserService.java b/omega/omega-spring-tx/src/test/java/org/apache/servicecomb/saga/omega/transaction/spring/TransactionalUserService.java
index 0a8cdfda9..c98c6ea5a 100644
--- a/omega/omega-spring-tx/src/test/java/org/apache/servicecomb/saga/omega/transaction/spring/TransactionalUserService.java
+++ b/omega/omega-spring-tx/src/test/java/org/apache/servicecomb/saga/omega/transaction/spring/TransactionalUserService.java
@@ -41,6 +41,6 @@ User add(User user) {
   }
 
   void delete(User user) {
-    userRepository.delete(user.id());
+    userRepository.delete(user);
   }
 }
diff --git a/pom.xml b/pom.xml
index 3a0194d04..3747ae1b8 100755
--- a/pom.xml
+++ b/pom.xml
@@ -47,6 +47,7 @@
     <kamon.version>0.6.7</kamon.version>
     <log4j.version>2.6.2</log4j.version>
     <spring.boot.version>1.5.10.RELEASE</spring.boot.version>
+    <spring.boot2.version>2.0.1.RELEASE</spring.boot2.version>
     <spring.cloud.version>Camden.SR6</spring.cloud.version>
     <java.chassis.version>0.5.0</java.chassis.version>
     <spring.version>4.3.7.RELEASE</spring.version>
diff --git a/saga-demo/pom.xml b/saga-demo/pom.xml
index 7b0e7340d..10f27dbd7 100644
--- a/saga-demo/pom.xml
+++ b/saga-demo/pom.xml
@@ -115,7 +115,7 @@
     <profile>
       <id>spring-boot-2</id>
       <properties>
-        <spring.boot.version>2.0.0.RELEASE</spring.boot.version>
+        <spring.boot.version>${spring.boot2.version}</spring.boot.version>
       </properties>
       <!-- Need to add the dependency of log4j2 core -->
       <dependencies>
diff --git a/saga-persistence/saga-persistence-jpa/pom.xml b/saga-persistence/saga-persistence-jpa/pom.xml
index f3c1cc2e5..7a5293012 100644
--- a/saga-persistence/saga-persistence-jpa/pom.xml
+++ b/saga-persistence/saga-persistence-jpa/pom.xml
@@ -89,7 +89,7 @@
     <profile>
       <id>spring-boot-2</id>
       <properties>
-        <spring.boot.version>2.0.0.RELEASE</spring.boot.version>
+        <spring.boot.version>${spring.boot2.version}</spring.boot.version>
       </properties>
     </profile>
   </profiles>