diff --git a/plot_single_transmission_EM9190.py b/plot_single_transmission_EM9190.py
index c09395b..cb6df52 100755
--- a/plot_single_transmission_EM9190.py
+++ b/plot_single_transmission_EM9190.py
@@ -136,7 +136,7 @@ if __name__ == "__main__":
 
         # Plot vertical lines
         first = True
-        lte_handovers = transmission_df["Cell_ID"].diff().dropna()
+        lte_handovers = transmission_df["Cell_ID"].dropna().diff()
         for index, value in lte_handovers.items():
             if value > 0:
                 if first:
@@ -146,7 +146,7 @@ if __name__ == "__main__":
                     ax00.axvline(index, ymin=0, ymax=1, color="skyblue")
 
         first = True
-        nr_handovers = transmission_df["NR5G_Cell_ID"].diff().dropna()
+        nr_handovers = transmission_df["NR5G_Cell_ID"].replace(0, np.NaN).dropna().diff()
         for index, value in nr_handovers.items():
             if value > 0:
                 if first: