diff --git a/lib/jekyll-seo-tag.rb b/lib/jekyll-seo-tag.rb
index 689e5b1..9f9b7e2 100644
--- a/lib/jekyll-seo-tag.rb
+++ b/lib/jekyll-seo-tag.rb
@@ -40,8 +40,8 @@ module Jekyll
 
     def payload
       {
-        "page"      => @context.registers[:page],
-        "site"      => @context.registers[:site].site_payload["site"],
+        "page"      => context.registers[:page],
+        "site"      => context.registers[:site].site_payload["site"],
         "paginator" => context["paginator"],
         "seo_tag"   => drop,
       }