diff --git a/labml_nn/transformers/basic/autoregressive_experiment.ipynb b/labml_nn/transformers/basic/autoregressive_experiment.ipynb
index e666c385..7daa2156 100644
--- a/labml_nn/transformers/basic/autoregressive_experiment.ipynb
+++ b/labml_nn/transformers/basic/autoregressive_experiment.ipynb
@@ -75,7 +75,7 @@
     "\n",
     "from labml import experiment\n",
     "from labml.configs import option\n",
-    "from labml_nn.transformers.basic.experiment import Configs"
+    "from labml_nn.transformers.basic.autoregressive_experiment import Configs"
    ],
    "execution_count": null,
    "outputs": []