transaction.rb in neo4j-10.0.0.pre.alpha.3

- old
+ new

@@ -20,11 +20,10 @@
         return yield(nil) unless run_in_tx
 
         tx = Neo4j::Transaction.new
         yield tx
       rescue Exception => e # rubocop:disable Lint/RescueException
-
         tx.mark_failed unless tx.nil?
         raise e
       ensure
         tx.close unless tx.nil?
       end
@@ -37,10 +36,10 @@
     def initialize(_options = {})
       (self.stack ||= []) << self
 
       @root = stack.first
       return unless root?
-      @driver_session = Neo4j::Core::Driver.singleton.driver.session(Neo4j::Driver::AccessMode::WRITE)
+      @driver_session = ActiveBase.current_driver.driver.session(Neo4j::Driver::AccessMode::WRITE)
       @driver_tx = @driver_session.begin_transaction
     rescue StandardError => e
       self.stack = []
       @driver_tx.close if @driver_tx
       @driver_session.close if @driver_session