diff --git a/src/aiidalab_qe_wannier90/result/utils.py b/src/aiidalab_qe_wannier90/result/utils.py
index 51daa62..5241136 100644
--- a/src/aiidalab_qe_wannier90/result/utils.py
+++ b/src/aiidalab_qe_wannier90/result/utils.py
@@ -77,7 +77,8 @@ def plot_skeaf(skeaf_data):
             # If neither phi nor theta has unique values, skip this band
             continue
 
-        labels.extend([band] * len(x))  # repeat band name for each point
+        # This is to ensure that frequencies from the same Wannier band have the same color in the plot
+        labels.extend([band] * len(freq))
 
     # Create scatter plot
     fig = px.scatter(x=x, y=y, color=labels,