lib/pg_partitioner.rb in pg_partitioner-0.4.1 vs lib/pg_partitioner.rb in pg_partitioner-0.4.2

@@ -24,10 +24,10 @@
   def partition_table_named_unique_indexes; end
 
   private
 
   def execute_sql(sql_string)
-    ActiveRecord::Base.connection.execute(sql_string)
+    connection.execute(sql_string)
   end
 
   def create_custom_index(table_name, index_fields, is_unique = false)
     ActiveRecord::Migration.add_index table_name, index_fields, unique: is_unique
   end