diff --git a/notebooks/DSSC/DSSC_Correct_and_Verify.ipynb b/notebooks/DSSC/DSSC_Correct_and_Verify.ipynb
index 5be08e8d43233e84db677ea027f5ce1f90e9becc..26983cedcee0a42f60a81d63366ab02dd5a91da0 100644
--- a/notebooks/DSSC/DSSC_Correct_and_Verify.ipynb
+++ b/notebooks/DSSC/DSSC_Correct_and_Verify.ipynb
@@ -465,8 +465,8 @@
     "                    if train == 0:\n",
     "                        pulseId = np.repeat(pulseId[:, None], data.shape[1], axis=1)\n",
     "                        pulseId = np.repeat(pulseId[:,:,None], data.shape[2], axis=2)\n",
-    "                        bins = (55, pulseId.max())\n",
-    "                        rnge = [[-5, 50], [0, pulseId.max()]]\n",
+    "                        bins = (55, int(pulseId.max()))\n",
+    "                        rnge = [[-5, 50], [0, int(pulseId.max())]]\n",
     "\n",
     "                        hists_signal_low, low_edges, pulse_edges = np.histogram2d(data.flatten(),\n",
     "                                                                                  pulseId.flatten(),\n",