diff --git a/plot_transmission_timeline.py b/plot_transmission_timeline.py
index 83a4b0d..0b10206 100755
--- a/plot_transmission_timeline.py
+++ b/plot_transmission_timeline.py
@@ -148,7 +148,7 @@ if __name__ == "__main__":
     print(transmission_df)
 
     # key for columns and level for index
-    transmission_df["goodput"] = transmission_df.groupby(pd.Grouper(level="datetime", freq="{}s".format(args.interval))).sum()
+    transmission_df["goodput"] = transmission_df["payload_size"].groupby(pd.Grouper(level="datetime", freq="{}s".format(args.interval))).sum()
     #transmission_df["goodput"] = transmission_df["payload_size"].rolling("{}s".format(args.interval)).sum()
     transmission_df["goodput"] = transmission_df["goodput"].apply(
         lambda x: ((x * 8) / args.interval) / 10**6