bin/console in unread-0.7.0 vs bin/console in unread-0.7.1

@@ -18,10 +18,10 @@
 # (If you use this, don't forget to add pry to your Gemfile!)
 # require "pry"
 # Pry.start
 
 ActiveRecord::Base.configurations = YAML.load_file('spec/database.yml')
-ActiveRecord::Base.establish_connection(:sqlite)
+ActiveRecord::Base.establish_connection((ENV['DB'] || 'sqlite').to_sym)
 ActiveRecord::Migration.verbose = false
 
 UnreadMigration.up
 SpecMigration.up