diff --git a/lib/assert_json/assert_json.rb b/lib/assert_json/assert_json.rb
index 1831d7e..176ee15 100644
--- a/lib/assert_json/assert_json.rb
+++ b/lib/assert_json/assert_json.rb
@@ -52,7 +52,7 @@ def item(index)
       decoded_json_in_scope = @decoded_json
       @decoded_json = @decoded_json[index]
       begin
-        yield if block_given?
+        yield @decoded_json if block_given?
         test_for_unexpected_keys(index)
       ensure
         @decoded_json = decoded_json_in_scope
@@ -116,7 +116,7 @@ def element(*args)
                         else
                           token
                         end
-        yield
+        yield @decoded_json
         test_for_unexpected_keys(arg)
       ensure
         @expected_keys = expected_keys_in_scope