diff --git a/main/astrodatagui/ui/FlaredetectorWidget.py b/main/astrodatagui/ui/FlaredetectorWidget.py
index c6fd3b9..0ecc4a0 100644
--- a/main/astrodatagui/ui/FlaredetectorWidget.py
+++ b/main/astrodatagui/ui/FlaredetectorWidget.py
@@ -196,7 +196,7 @@ class FlaredetectorWidget(QtWidgets.QWidget):
             lc = lc.to_periodogram(method=self.PeriodogramState["Method"])
             label = "Periodogram"
 
-            maxPowers = findMaxIndices(lc.power, 4)
+            maxPowers = findMaxIndices(lc, 4)
             print(maxPowers)
             self.lbMaxPowerPeriod1.setText(str(lc.period[maxPowers[0]]))
             self.lbMaxPowerPeriod2.setText(str(lc.period[maxPowers[1]]))
diff --git a/main/flaredetector/util.py b/main/flaredetector/util.py
index 52163e2..c0dfcc6 100644
--- a/main/flaredetector/util.py
+++ b/main/flaredetector/util.py
@@ -1,8 +1,11 @@
 import numpy as np
 from scipy.signal import find_peaks
 
-def findMaxIndices(data, num=3, distance=100):
-    peak_indices, peak_dict = find_peaks(data, height=(None, None), distance=distance)
-    peak_heights = peak_dict["peak_heights"]
-    maxIndices = peak_indices[np.argsort(peak_heights)[-num:][::-1]]
-    return maxIndices
\ No newline at end of file
+from lightkurve.periodogram import Periodogram
+
+def findMaxIndices(lc, num=3, distance=100):
+    if(isinstance(lc, Periodogram)):
+        peak_indices, peak_dict = find_peaks(lc.power, height=(None, None), distance=distance)
+        peak_heights = peak_dict["peak_heights"]
+        maxIndices = peak_indices[np.argsort(peak_heights)[-num:][::-1]]
+        return maxIndices
\ No newline at end of file