diff --git a/test/deep_pluck_test.rb b/test/deep_pluck_test.rb
index <HASH>..<HASH> 100644
--- a/test/deep_pluck_test.rb
+++ b/test/deep_pluck_test.rb
@@ -210,7 +210,12 @@ class DeepPluckTest < Minitest::Test
     assert_equal post_expected, Post.where(id: 1).deep_pluck(:notes => [:content])
   end
 
-  def test_at_model
+  def test_pluck_at_model_with_1_level_deep
+    expected = {'name' => 'Pearl' }
+    assert_equal(expected, User.where(:name => %w(Pearl)).first.deep_pluck(:name))
+  end
+
+  def test_pluck_at_model_with_2_level_deep
     expected = {'name' => 'Pearl' , :posts => [{'name' => "post4"}, {'name' => "post5"}]}
     assert_equal(expected, User.where(:name => %w(Pearl)).first.deep_pluck(:name, :posts => [:name]))
   end
test #deep_pluck at model without plucking associations