diff --git a/gateware/led.py b/gateware/led.py
index c930c6d..8f2fc7e 100644
--- a/gateware/led.py
+++ b/gateware/led.py
@@ -1,5 +1,6 @@
 from amaranth import *
 from amaranth_soc.wishbone import *
+from amaranth_soc.memory import *
 
 
 class LEDPeripheral(Elaboratable, Interface):
diff --git a/gateware/main.py b/gateware/main.py
index 7672d95..388fe33 100644
--- a/gateware/main.py
+++ b/gateware/main.py
@@ -10,6 +10,9 @@ from minerva.core import Minerva
 from typing import List
 from argparse import ArgumentParser
 
+from memory import *
+from led import *
+
 
 # To change clock domain of a module:
 # new_thing = DomainRenamer("new_clock")(MyElaboratable())