diff --git a/test/config.py b/test/config.py
index 5d367ccd3..60687bf3c 100644
--- a/test/config.py
+++ b/test/config.py
@@ -6,9 +6,9 @@ busword = 32
 debug_wb = False
 
 if use_uart:
-	from litescope.host.driver import LiteScopeUART2WBDriver
+	from litescope.host.driver.uart import LiteScopeUARTDriver
 	wb = LiteScopeUART2WBDriver(2, 921600, csr_csv_file, busword, debug_wb)
 
 if use_eth:
-	from litescope.host.driver import LiteScopeEtherboneDriver
+	from litescope.host.driver.etherbone import LiteScopeEtherboneDriver
 	wb = LiteScopeEtherboneDriver("192.168.1.40", 20000, csr_csv_file, debug_wb)
diff --git a/test/test_la.py b/test/test_la.py
index fe5d20196..56bf302fa 100644
--- a/test/test_la.py
+++ b/test/test_la.py
@@ -1,7 +1,7 @@
 from config import *
 import time
 
-from litescope.host.driver import LiteScopeLADriver
+from litescope.host.driver.la import LiteScopeLADriver
 la = LiteScopeLADriver(wb.regs, "la", debug=True)
 
 wb.open()