diff --git a/docs/distillation/index.html b/docs/distillation/index.html
index c93d86a2d72568d3dda72e3d72319100cc7d9ee6..efc685307718f51e25709715c05140846f782351 100644
--- a/docs/distillation/index.html
+++ b/docs/distillation/index.html
@@ -324,7 +324,7 @@ by a factor of $T^2$</p>
 </p>
             </div>
             <div class='code'>
-                <div class="highlight"><pre><span class="lineno">143</span>        <span class="n">soft_prob</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">functional</span><span class="o">.</span><span class="n">log_softmax</span><span class="p">(</span><span class="n">large_logits</span> <span class="o">/</span> <span class="bp">self</span><span class="o">.</span><span class="n">temperature</span><span class="p">,</span> <span class="n">dim</span><span class="o">=-</span><span class="mi">1</span><span class="p">)</span></pre></div>
+                <div class="highlight"><pre><span class="lineno">143</span>        <span class="n">soft_prob</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">functional</span><span class="o">.</span><span class="n">log_softmax</span><span class="p">(</span><span class="n">output</span> <span class="o">/</span> <span class="bp">self</span><span class="o">.</span><span class="n">temperature</span><span class="p">,</span> <span class="n">dim</span><span class="o">=-</span><span class="mi">1</span><span class="p">)</span></pre></div>
             </div>
         </div>
     <div class='section' id='section-18'>
diff --git a/docs/sitemap.xml b/docs/sitemap.xml
index adf02315f25a23453ffe9758bf9ebe484988e9c8..6cda81c07e6e003c9c66882bd97c36a638067acf 100644
--- a/docs/sitemap.xml
+++ b/docs/sitemap.xml
@@ -316,7 +316,7 @@
 
     <url>
       <loc>https://nn.labml.ai/distillation/index.html</loc>
-      <lastmod>2021-07-03T16:30:00+00:00</lastmod>
+      <lastmod>2021-07-12T16:30:00+00:00</lastmod>
       <priority>1.00</priority>
     </url>
     
@@ -337,7 +337,7 @@
 
     <url>
       <loc>https://nn.labml.ai/index.html</loc>
-      <lastmod>2021-07-03T16:30:00+00:00</lastmod>
+      <lastmod>2021-07-08T16:30:00+00:00</lastmod>
       <priority>1.00</priority>
     </url>
     
@@ -834,7 +834,7 @@
 
     <url>
       <loc>https://nn.labml.ai/graphs/index.html</loc>
-      <lastmod>2021-07-07T16:30:00+00:00</lastmod>
+      <lastmod>2021-07-08T16:30:00+00:00</lastmod>
       <priority>1.00</priority>
     </url>