diff --git a/zxlive/edit_panel.py b/zxlive/edit_panel.py
index ffc2608..b3a70c4 100644
--- a/zxlive/edit_panel.py
+++ b/zxlive/edit_panel.py
@@ -12,7 +12,7 @@
 
 from .base_panel import ToolbarSection
 from .commands import UpdateGraph
-from .common import GraphT
+from .common import GraphT, get_data
 from .dialogs import show_error_msg, create_circuit_dialog
 from .editor_base_panel import EditorBasePanel
 from .graphscene import EditGraphScene