diff --git a/Python_script/prototype.py b/Python_script/prototype.py
index 25d6fa1516666fbad882ea1c4ae9f13c2c51bb4b..28eac2fe0d8a683116dc0ccd9dfbb3e0ce83e200 100755
--- a/Python_script/prototype.py
+++ b/Python_script/prototype.py
@@ -62,6 +62,7 @@ class Measurements:
         self.vna.create_new_trace("Trace4", "S22")
 
         self.measurement_plot = MeasurementPlot.MeasurementPlot()
+        self.data_collection = []
 
     def perform_measurements(self):
         with open(self.output, mode='w', newline='') as csv_file:
@@ -191,6 +192,8 @@ class Measurements:
 
                                 writer.writerow(measurement)
 
+                    plt.close()
+
                 except KeyboardInterrupt:
                     pass