diff --git a/plot_single_transmission.py b/plot_single_transmission.py
index 029eec4..cfd28d9 100755
--- a/plot_single_transmission.py
+++ b/plot_single_transmission.py
@@ -95,11 +95,11 @@ if __name__ == "__main__":
 
             scaley = 1.5
             scalex = 1.0
-            ax0 = plt.subplots(211, figsize=[6.4 * scaley, 4.8 * scalex])
+            fig, ax0 = plt.subplots(211, figsize=[6.4 * scaley, 4.8 * scalex])
             ax1 = ax0.twinx()
             ax2 = ax0.twinx()
 
-            ax00 = plt.subplots(212)
+            fig, ax00 = plt.subplots(212)
             ax01 = ax00.twinx()
 
             plt.title("{} with {}".format(transmission_direction, cc_algo))