diff --git a/docs/io/visualization/montecarlo_packet_visualization.ipynb b/docs/io/visualization/montecarlo_packet_visualization.ipynb
index ebd21756299..62b80f30871 100644
--- a/docs/io/visualization/montecarlo_packet_visualization.ipynb
+++ b/docs/io/visualization/montecarlo_packet_visualization.ipynb
@@ -23,6 +23,7 @@
    "source": [
     "from tardis import run_tardis\n",
     "from tardis.io.config_reader import Configuration\n",
+    "from tardis.io.atom_data.util import download_atom_data\n",
     "import plotly.express as px\n",
     "import plotly.graph_objects as go\n",
     "import math\n",
@@ -31,6 +32,16 @@
     "import random"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Download atom data\n",
+    "download_atom_data('kurucz_cd23_chianti_H_He')"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,