diff --git a/main/astrodatagui/ui/FlaredetectorWidget.py b/main/astrodatagui/ui/FlaredetectorWidget.py
index 6261261..15dc5a0 100644
--- a/main/astrodatagui/ui/FlaredetectorWidget.py
+++ b/main/astrodatagui/ui/FlaredetectorWidget.py
@@ -24,6 +24,7 @@ class FlaredetectorWidget(QtWidgets.QWidget):
         print(f"Plotting: {fitsFilePath}")
         self.figureAxis.clear()
         lc = lk.read(fitsFilePath)
+        lc.flux = lc['sap_flux']
         label = f"{mainName} - SAP Flux"
         lc.plot(column="sap_flux", label=label, ax=self.figureAxis)
         self.figure.canvas.draw_idle()
\ No newline at end of file