Rakefile in devise-neo4j-2.0.1 vs Rakefile in devise-neo4j-2.0.2

@@ -1,8 +1,13 @@
 # encoding: UTF-8
 require "bundler/gem_tasks"
 require 'rake/testtask'
 require 'rdoc/task'
+begin
+  require 'neo4j/rake_tasks'
+rescue LoadError
+  load 'neo4j/tasks/neo4j_server.rake'
+end
 
 task :default => [:test, :"neo4j:db:remove"]
 
 ENV['DEVISE_ORM'] = 'neo4j'
 devise_checked_out = File.join(File.dirname(__FILE__), '../devise')