diff --git a/tests/test_db_api.rb b/tests/test_db_api.rb
index c0a5a0d..3219a87 100644
--- a/tests/test_db_api.rb
+++ b/tests/test_db_api.rb
@@ -447,8 +447,8 @@ class DBAPITest < Test::Unit::TestCase
     @@coll.insert('a' => 3)
 
     assert_equal 3, @@coll.count
-    assert_equal 1, @@coll.find('$where' => Code.new('this.a > 2')).count
-    assert_equal 2, @@coll.find('$where' => Code.new('this.a > i', {'i' => 1})).count
+    assert_equal 1, @@coll.count('$where' => Code.new('this.a > 2'))
+    assert_equal 2, @@coll.count('$where' => Code.new('this.a > i', {'i' => 1}))
   end
 
   def test_hint