diff --git a/notebooks/LPD/LPD_Correct_Fast.ipynb b/notebooks/LPD/LPD_Correct_Fast.ipynb
index 6c6990439238b4c07a9aeef400843380a090afed..8769da543eecdc0ccf1c9543dca10d8e8f8f47a5 100644
--- a/notebooks/LPD/LPD_Correct_Fast.ipynb
+++ b/notebooks/LPD/LPD_Correct_Fast.ipynb
@@ -61,12 +61,13 @@
     "\n",
     "# Parallelization options\n",
     "sequences_per_node = 1  # Sequence files to process per node\n",
+    "max_nodes = 8  # Maximum number of SLURM jobs to split correction work into\n",
     "num_workers = 8  # Worker processes per node, 8 is safe on 768G nodes but won't work on 512G.\n",
     "num_threads_per_worker = 32  # Number of threads per worker.\n",
     "\n",
-    "def balance_sequences(in_folder, run, sequences, sequences_per_node, karabo_da):\n",
+    "def balance_sequences(in_folder, run, sequences, sequences_per_node, karabo_da, max_nodes):\n",
     "    from xfel_calibrate.calibrate import balance_sequences as bs\n",
-    "    return bs(in_folder, run, sequences, sequences_per_node, karabo_da)"
+    "    return bs(in_folder, run, sequences, sequences_per_node, karabo_da, max_nodes=max_nodes)"
    ]
   },
   {