diff --git a/example_designs/test/test_logic_analyzer.py b/example_designs/test/test_logic_analyzer.py
index 889f148..205d60a 100644
--- a/example_designs/test/test_logic_analyzer.py
+++ b/example_designs/test/test_logic_analyzer.py
@@ -1,4 +1,5 @@
 from litex.soc.tools.remote import RemoteClient
+from litescope.software.driver.logic_analyzer import LiteScopeLogicAnalyzerDriver
 
 wb = RemoteClient()
 wb.open()