diff --git a/doc/BOZ analysis part II.1 Small data.ipynb b/doc/BOZ analysis part II.1 Small data.ipynb
index b2c1a8f698e837c1137b9c842cb86eb781a76efe..eac18f066175fb4a5851063723ca84443a2410a1 100644
--- a/doc/BOZ analysis part II.1 Small data.ipynb	
+++ b/doc/BOZ analysis part II.1 Small data.ipynb	
@@ -125,7 +125,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "arr_dark, tid_dark = boz.load_dssc_module(proposal, darkrunNB, moduleNB)"
+    "arr_dark, tid_dark = boz.load_dssc_module(proposal, darkrunNB, moduleNB)",
+    "arr_dark = arr_dark.rechunk(('auto',-1,-1,-1))\n"
    ]
   },
   {
@@ -175,6 +176,7 @@
    "source": [
     "dark = boz.average_module(arr_dark).compute()\n",
     "arr, tid = boz.load_dssc_module(proposal, runNB, moduleNB)\n",
+    "arr = arr.rechunk(('auto',-1,-1,-1))\n",
     "data = boz.average_module(arr, dark=dark).compute()\n",
     "pp = data.mean(axis=(1,2)) # pulseId resolved mean\n",
     "dataM = data.mean(axis=0) # mean over pulseId"
@@ -370,14 +372,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# load data\n",
-    "arr, tid = boz.load_dssc_module(proposal, runNB, moduleNB)\n",
-    "arr_dark, tid_dark = boz.load_dssc_module(proposal, darkrunNB, moduleNB)\n",
-    "\n",
-    "# make sure to rechunk the arrays\n",
-    "arr = arr.rechunk(('auto',-1,-1,-1))\n",
-    "arr_dark = arr_dark.rechunk(('auto',-1,-1,-1))\n",
-    "\n",
     "#no correction\n",
     "#data = boz.process(np.arange(2**9), arr_dark, arr, tid, rois, params.get_mask(),\n",
     "#                   np.ones_like(ff), params.sat_level)\n",