diff --git a/data/generate_dataset.py b/data/generate_dataset.py
index 67ab95e..f106237 100644
--- a/data/generate_dataset.py
+++ b/data/generate_dataset.py
@@ -125,9 +125,7 @@ def main(args):
                                            (rise_size[id_idx] >= 10)], dtype=int)
                 ax.plot(times_v[rise_idx_oi], fish_freq[id_idx][rise_idx_oi], 'o', color='tab:red')
 
-            plt.show(block=False)
-            time.sleep(10)
-            plt.close()
+            plt.show()
 
 
 if __name__ == '__main__':