diff --git a/notebooks/single_baseline_2D_filtered_SNRs.ipynb b/notebooks/single_baseline_2D_filtered_SNRs.ipynb
index 609b3cf..175fcef 100644
--- a/notebooks/single_baseline_2D_filtered_SNRs.ipynb
+++ b/notebooks/single_baseline_2D_filtered_SNRs.ipynb
@@ -188,7 +188,7 @@
     "    \n",
     "    axes[0].set_ylabel(f'JD - {int(data.times[0])}')\n",
     "    plt.tight_layout()\n",
-    "    largest_pixel = np.max([np.abs(filtered_SNR[bl][~flags[bl]]) \n",
+    "    largest_pixel = np.max([np.max(np.abs(filtered_SNR[bl][~flags[bl]])) \n",
     "                            for bl in filtered_SNR if not np.all(flags[bl])])\n",
     "    plt.colorbar(im, ax=axes, label='|2D DPSS Filtered SNR|', pad=.02, \n",
     "                 extend=('max' if largest_pixel > vmax else None))\n",