bug: fix broken basic table arrows on other widgets

1 files changed, 10 insertions, 1 deletions

diff --git a/src/canvas.rs b/src/canvas.rs
index 9c598567..5f243852 100644
--- a/src/canvas.rs
+++ b/src/canvas.rs
@@ -386,9 +386,18 @@ impl Painter {
                 self.draw_basic_cpu(&mut f, app_state, vertical_chunks[0], 1);
                 self.draw_basic_memory(&mut f, app_state, middle_chunks[0], 2);
                 self.draw_basic_network(&mut f, app_state, middle_chunks[1], 3);
-                self.draw_basic_table_arrows(&mut f, app_state, vertical_chunks[3]);
                 if let Some(basic_table_widget_state) = &app_state.basic_table_widget_state {
                     let widget_id = basic_table_widget_state.currently_displayed_widget_id;
+
+                    if let Some(current_table) = app_state.widget_map.get(&widget_id) {
+                        self.draw_basic_table_arrows(
+                            &mut f,
+                            app_state,
+                            vertical_chunks[3],
+                            current_table,
+                        );
+                    }
+
                     match basic_table_widget_state.currently_displayed_widget_type {
                         Disk => self.draw_disk_table(
                             &mut f,