diff --git a/plot_transmission_timeline.py b/plot_transmission_timeline.py
index 54405c4..f13ea92 100755
--- a/plot_transmission_timeline.py
+++ b/plot_transmission_timeline.py
@@ -189,11 +189,10 @@ if __name__ == "__main__":
 
     host = host_subplot(111, axes_class=axisartist.Axes)
 
-    #cmap = matplotlib.cm.get_cmap("Set3")
-    #for c in transmission_df["cellID"].unique():
-    #    bounds = transmission_df[["cellID"]].groupby("cellID").agg(["min", "max"]).loc[c]
-    #    host.axvspan(bounds.min(), bounds.max() + 1, alpha=0.3, color=cmap.colors[c])
-
+    cmap = matplotlib.cm.get_cmap("Set3")
+    for c in transmission_df["cellID"].unique():
+        bounds = transmission_df[[transmission_df.index, "cellID"]].groupby("cellID").agg(["min", "max"]).loc[c]
+        host.axvspan(bounds.min(), bounds.max() + 1, alpha=0.3, color=cmap.colors[c])
 
     plt.subplots_adjust()