diff --git a/src/wela/dataloader.py b/src/wela/dataloader.py
index 76ce136ff132fcf5008dc9fd76a426079174b55c..85a6679c4ad1d3e0a64ff42cf4f31f65e301540b 100644
--- a/src/wela/dataloader.py
+++ b/src/wela/dataloader.py
@@ -189,7 +189,7 @@ class dataloader:
         self.a2g_dict = {v: k for (k, v) in self.g2a_dict.items()}
 
     def include_bud_fluorescence(self, grouper, dataname):
-        """Add mean and median bud fluorescence to the h5 files."""
+        """Add statistics for bud fluorescence to the h5 files."""
         # find fluorescence channels
         channels = list(grouper.channels)
         channels.remove("Brightfield")
@@ -199,6 +199,8 @@ class dataloader:
                 [
                     f"/extraction/{channel}/max/median",
                     f"/extraction/{channel}/max/mean",
+                    f"/extraction/{channel}/max/total",
+                    f"/extraction/{channel}/max/total_squared",
                 ]
                 for channel in channels
             ]