diff --git a/notebooks/file_calibration.ipynb b/notebooks/file_calibration.ipynb
index 549af43..b98f315 100644
--- a/notebooks/file_calibration.ipynb
+++ b/notebooks/file_calibration.ipynb
@@ -519,7 +519,7 @@
     "    avg_auto = {(-1, -1, 'ee'): np.mean([data[bl] for bl in auto_bls_to_use], axis=0)}\n",
     "    \n",
     "    # Flag RFI first with channel differences and then with DPSS\n",
-    "    antenna_flags, _ = xrfi.flag_autos(avg_auto, int_count=int_count, nsig=(RFI_NSIG * 5))\n",
+    "    antenna_flags, _ = xrfi.flag_autos(avg_auto, int_count=int_count, nsig=(nsig * 5))\n",
     "    if prior_flags is not None:\n",
     "        antenna_flags[(-1, -1, 'ee')] = prior_flags\n",
     "    _, rfi_flags = xrfi.flag_autos(avg_auto, int_count=int_count, flag_method='dpss_flagger',\n",