diff --git a/docs/experiments/nlp_autoregression.html b/docs/experiments/nlp_autoregression.html
index e35de1a6..5b0ffb0b 100644
--- a/docs/experiments/nlp_autoregression.html
+++ b/docs/experiments/nlp_autoregression.html
@@ -83,7 +83,7 @@
 <span class="lineno">19</span><span class="kn">from</span> <span class="nn">labml.logger</span> <span class="kn">import</span> <span class="n">Text</span>
 <span class="lineno">20</span><span class="kn">from</span> <span class="nn">labml_helpers.datasets.text</span> <span class="kn">import</span> <span class="n">TextDataset</span><span class="p">,</span> <span class="n">SequentialDataLoader</span><span class="p">,</span> <span class="n">SequentialUnBatchedDataset</span><span class="p">,</span> <span class="n">TextFileDataset</span>
 <span class="lineno">21</span><span class="kn">from</span> <span class="nn">labml_helpers.device</span> <span class="kn">import</span> <span class="n">DeviceConfigs</span>
-<span class="lineno">22</span><span class="kn">from</span> <span class="nn">labml_helpers.metrics.accuracy</span> <span class="kn">import</span> <span class="n">Accuracy</span>
+<span class="lineno">22</span><span class="kn">from</span> <span class="nn">labml_helpers.metrics.accuracy</span> <span class="kn">import</span> <span class="n">Accuracy</span><span class="p">,</span> <span class="n">AccuracyMovingAvg</span>
 <span class="lineno">23</span><span class="kn">from</span> <span class="nn">labml_helpers.module</span> <span class="kn">import</span> <span class="n">Module</span>
 <span class="lineno">24</span><span class="kn">from</span> <span class="nn">labml_helpers.train_valid</span> <span class="kn">import</span> <span class="n">TrainValidConfigs</span><span class="p">,</span> <span class="n">hook_model_outputs</span><span class="p">,</span> <span class="n">BatchIndex</span>
 <span class="lineno">25</span><span class="kn">from</span> <span class="nn">labml_nn.optimizers.configs</span> <span class="kn">import</span> <span class="n">OptimizerConfigs</span></pre></div>
@@ -293,7 +293,7 @@
 
         </div>
         <div class='code'>
-            <div class="highlight"><pre><span class="lineno">80</span>    <span class="n">accuracy</span> <span class="o">=</span> <span class="n">Accuracy</span><span class="p">()</span></pre></div>
+            <div class="highlight"><pre><span class="lineno">80</span>    <span class="n">accuracy</span> <span class="o">=</span> <span class="n">AccuracyMovingAvg</span><span class="p">()</span></pre></div>
         </div>
     </div>
     <div class='section' id='section-18'>
diff --git a/setup.py b/setup.py
index c923f274..2c01dc17 100644
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,7 @@ setuptools.setup(
                                                'test',
                                                'test.*')),
     install_requires=['labml>=0.4.151',
-                      'labml-helpers>=0.4.86',
+                      'labml-helpers>=0.4.87',
                       'torch',
                       'torchtext',
                       'torchvision',