lib/second_base/on_base.rb in secondbase-2.1.3 vs lib/second_base/on_base.rb in secondbase-2.1.4

@@ -8,10 +8,10 @@
     if already_on_base
       yield
       return
     end
     original_config = ActiveRecord::Tasks::DatabaseTasks.current_config
-    original_configurations = ActiveRecord::Base.configurations
+    original_configurations = Rails.application.config.database_configuration
     original_migrations_path = ActiveRecord::Tasks::DatabaseTasks.migrations_paths
     original_db_dir = ActiveRecord::Tasks::DatabaseTasks.db_dir
     ActiveRecord::Tasks::DatabaseTasks.current_config = config
     ActiveRecord::Base.configurations = original_configurations[Railtie.config_key]
     ActiveRecord::Base.establish_connection(config)