diff --git a/b_asic/scheduler_gui/__init__.py b/b_asic/scheduler_gui/__init__.py
index 5c558d3445b4b0bae91f4a168dfb9e7a83d36980..d8ebf20836bd9d9d8b457ed86f621cc6dc01cf6e 100644
--- a/b_asic/scheduler_gui/__init__.py
+++ b/b_asic/scheduler_gui/__init__.py
@@ -9,6 +9,7 @@ from graphics_graph_item        import *
 from graphics_graph_event       import *
 from graphics_component_item    import *
 from graphics_axes_item         import *
+from graphics_signal            import *
 from graphics_timeline_item     import *
 
 # __all__ = ['main_window', 'graphics_graph', 'component_item', 'graphics_axes', 'graphics_timeline_item']
diff --git a/test/test_scheduler_gui.py b/test/test_scheduler_gui.py
index 0085fc7fb3ef78be599dad65644a86d79e712b4a..4d6a41d9d6ba38b4b2292aeafb10379d466321b4 100644
--- a/test/test_scheduler_gui.py
+++ b/test/test_scheduler_gui.py
@@ -1,7 +1,7 @@
 import pytest
 
 try:
-    import b_asic.scheduler_gui.main_window as GUI
+    import b_asic.scheduler_gui as GUI
 except ImportError:
     pytestmark = pytest.mark.skip('Qt not setup')