1 files changed, 8 insertions, 8 deletions
diff --git a/activerecord/test/cases/adapters/postgresql/range_test.rb b/activerecord/test/cases/adapters/postgresql/range_test.rb
index b57568dbad..57c7da2657 100644
--- a/activerecord/test/cases/adapters/postgresql/range_test.rb
+++ b/activerecord/test/cases/adapters/postgresql/range_test.rb
@@ -1,5 +1,5 @@
 require "cases/helper"
-require 'support/postgresql_helper'
+require 'support/connection_helper'
 require 'active_record/base'
 require 'active_record/connection_adapters/postgresql_adapter'
 
@@ -10,13 +10,7 @@ if ActiveRecord::Base.connection.supports_ranges?
 
   class PostgresqlRangeTest < ActiveRecord::TestCase
     self.use_transactional_fixtures = false
-    include PostgresqlHelper
-
-    teardown do
-      @connection.execute 'DROP TABLE IF EXISTS postgresql_ranges'
-      @connection.execute 'DROP TYPE IF EXISTS floatrange'
-      reset_pg_session
-    end
+    include ConnectionHelper
 
     def setup
       @connection = PostgresqlRange.connection
@@ -99,6 +93,12 @@ _SQL
       @empty_range = PostgresqlRange.find(105)
     end
 
+    teardown do
+      @connection.execute 'DROP TABLE IF EXISTS postgresql_ranges'
+      @connection.execute 'DROP TYPE IF EXISTS floatrange'
+      reset_connection
+    end
+
     def test_data_type_of_range_types
       assert_equal :daterange, @first_range.column_for_attribute(:date_range).type
       assert_equal :numrange, @first_range.column_for_attribute(:num_range).type