Rakefile in token_action-0.0.1 vs Rakefile in token_action-0.0.2

@@ -1,11 +1,20 @@
-require 'bundler'
-Bundler::GemHelper.install_tasks
+begin
+  require 'bundler/setup'
+rescue LoadError
+  puts 'You must `gem install bundler` and `bundle install` to run rake tasks'
+end
 
+APP_RAKEFILE = File.expand_path('../spec/dummy/Rakefile', __FILE__)
+load 'rails/tasks/engine.rake'
+
+require 'thor'
+helper = Bundler::GemHelper.new
+helper.install
+
 require 'rspec/core/rake_task'
 RSpec::Core::RakeTask.new(:spec)
-
-task :default => :spec
+task default: :spec
 
 begin
   require 'yard'
   YARD::Rake::YardocTask.new
 rescue LoadError