diff --git a/Gemfile b/Gemfile
index f73244f..aa56ca9 100644
--- a/Gemfile
+++ b/Gemfile
@@ -5,3 +5,4 @@ gemspec
 
 # gem "mongoid", github: "mongoid/mongoid"
 # gem "mongoid", "~> 3.1.0"
+# gem "activerecord", "~> 3.2.0"
diff --git a/test/similar_test.rb b/test/similar_test.rb
index 39c126a..39daad8 100644
--- a/test/similar_test.rb
+++ b/test/similar_test.rb
@@ -4,7 +4,7 @@ class TestSimilar < Minitest::Unit::TestCase
 
   def test_fields
     store_names ["1% Organic Milk", "2% Organic Milk", "Popcorn"]
-    assert_equal ["2% Organic Milk"], Product.find_by(name: "1% Organic Milk").similar(fields: ["name"]).map(&:name)
+    assert_equal ["2% Organic Milk"], Product.where(name: "1% Organic Milk").first.similar(fields: ["name"]).map(&:name)
   end
 
 end
diff --git a/test/test_helper.rb b/test/test_helper.rb
index 0ceefca..156add2 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -66,7 +66,7 @@ class Product < ActiveRecord::Base
   attr_accessor :conversions, :user_ids
 
   def search_data
-    as_json.merge conversions: conversions, user_ids: user_ids
+    as_json(root: false).merge conversions: conversions, user_ids: user_ids
   end
 end
 
--
libgit2 0.21.0