From 0514eb7eb1d994ff053347ca6fe0200f8f9fe478 Mon Sep 17 00:00:00 2001
From: timerke <timerke@mail.ru>
Date: Thu, 14 Dec 2023 10:59:06 +0300
Subject: [PATCH] Fixed legend

---
 report_generator/plot.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/report_generator/plot.py b/report_generator/plot.py
index fd5c31b..d4932e6 100644
--- a/report_generator/plot.py
+++ b/report_generator/plot.py
@@ -165,6 +165,12 @@ def draw_ivc_for_pin(pin_info: PinInfo, index: int, file_name: str, scaling_type
     else:
         test_curve.clear_curve()
 
+    for curve in viewer.plot.curves:
+        if curve.curve is None:
+            curve.detach()
+        else:
+            curve.attach(viewer.plot)
+
     viewer.plot.grab().save(file_name, format="PNG")