diff --git a/beta_spectrum/visualize/__init__.py b/beta_spectrum/visualize/__init__.py
index 30763e2..2b32e9a 100644
--- a/beta_spectrum/visualize/__init__.py
+++ b/beta_spectrum/visualize/__init__.py
@@ -742,14 +742,14 @@ class BetaSpectrumAnalyzer:
                 neutrino_mass_keV = self.config.neutrino_mass_MeV * 1000
                 if branch_endpoint_keV <= neutrino_mass_keV:
                     # Branch is degenerate, spectrum should be all zeros
-                    data[f"branch_{i + 1}_spectrum_normalized"] = np.zeros_like(branch_spectrum)
+                    data[f"branch_{i + 1}"] = np.zeros_like(branch_spectrum)
                 else:
                     # Branch is valid, normalize as usual
                     sum_spectrum = np.sum(branch_spectrum)
                     if sum_spectrum > 0:
-                        data[f"branch_{i + 1}_spectrum_normalized"] = branch_spectrum / sum_spectrum
+                        data[f"branch_{i + 1}"] = branch_spectrum / sum_spectrum
                     else:
-                        data[f"branch_{i + 1}_spectrum_normalized"] = branch_spectrum
+                        data[f"branch_{i + 1}"] = branch_spectrum
 
         df = pd.DataFrame(data)
         df.to_csv(filename, index=False, float_format="%.4e")