diff --git a/main/astrodatagui/ui/FlaredetectorWidget.py b/main/astrodatagui/ui/FlaredetectorWidget.py
index 00e6660..4562e98 100644
--- a/main/astrodatagui/ui/FlaredetectorWidget.py
+++ b/main/astrodatagui/ui/FlaredetectorWidget.py
@@ -148,12 +148,13 @@ class FlaredetectorWidget(QtWidgets.QWidget):
         self.updatePlot()
 
     def normalizeStichedLightCurve(self, lc):
-        print(lc, self.NormalizeState["Scale"])
+        lc.flux = lc[self.fluxType]
         return lc.normalize(unit=self.NormalizeState["Scale"])
 
     def updatePlot(self):
         lc = self.currentLC
         label = f"{self.currentMainName}"
+        lc.flux = lc[self.fluxType]
         if(self.NormalizeState["Enabled"]):
             if(self.currentLCCollection is None):
                 print("normalizing normal lc")
@@ -162,7 +163,6 @@ class FlaredetectorWidget(QtWidgets.QWidget):
                 print("normalizing collection")
                 lc = self.currentLCCollection.stitch(self.normalizeStichedLightCurve)
             label += " - normalized"
-        lc.flux = lc[self.fluxType]
         if(self.RemoveOutliersState["Enabled"]):
             lc = lc.remove_outliers(sigma=self.RemoveOutliersState["Sigma"])
             label += " - no ol"