diff --git a/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSpringTransaction.java b/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSpringTransaction.java
index cb650ebf..59e03d25 100644
--- a/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSpringTransaction.java
+++ b/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSpringTransaction.java
@@ -20,6 +20,7 @@ import com.mybatisflex.core.transaction.TransactionContext;
 import com.mybatisflex.core.util.StringUtil;
 import org.apache.ibatis.transaction.Transaction;
 
+import javax.sql.DataSource;
 import java.sql.Connection;
 import java.sql.SQLException;
 
@@ -31,12 +32,12 @@ import java.sql.SQLException;
  */
 public class FlexSpringTransaction implements Transaction {
 
-    private final FlexDataSource dataSource;
+    private final DataSource dataSource;
     private Boolean isConnectionTransactional;
     private Boolean autoCommit;
     private Connection connection;
 
-    public FlexSpringTransaction(FlexDataSource dataSource) {
+    public FlexSpringTransaction(DataSource dataSource) {
         this.dataSource = dataSource;
     }
 
diff --git a/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexTransactionFactory.java b/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexTransactionFactory.java
index 9e3b83f2..a6c92841 100644
--- a/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexTransactionFactory.java
+++ b/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexTransactionFactory.java
@@ -34,7 +34,7 @@ public class FlexTransactionFactory extends SpringManagedTransactionFactory {
      */
     @Override
     public Transaction newTransaction(DataSource dataSource, TransactionIsolationLevel level, boolean autoCommit) {
-        return new FlexSpringTransaction((FlexDataSource) dataSource);
+        return new FlexSpringTransaction(dataSource);
     }
 
     /**