diff --git a/unoptimized_lr/simple_lr_automaton.py b/unoptimized_lr/simple_lr_automaton.py
index ff79e2ca0d8f656bfc14184f7d07eba1975b2c50..28ca4c36f3f466ad5d070d0ae652cc0064927cbb 100644
--- a/unoptimized_lr/simple_lr_automaton.py
+++ b/unoptimized_lr/simple_lr_automaton.py
@@ -2,6 +2,10 @@ from nmigen import *
 from nmigen.hdl.rec import *
 from nmigen.cli import main
 
+import rtl_lib
+import simple_lr_tables
+import simple_lr_stack
+import simple_parse_tree
 
 from functools import reduce
 
diff --git a/unoptimized_lr/simple_parse_tree.py b/unoptimized_lr/simple_parse_tree.py
index 2e03063bc86f642d3759df7ab4402405160497a4..f6c6b52182827fb3a224ce50b4e34121abf00b96 100644
--- a/unoptimized_lr/simple_parse_tree.py
+++ b/unoptimized_lr/simple_parse_tree.py
@@ -1,3 +1,8 @@
+from nmigen import *
+from nmigen.hdl.rec import *
+from nmigen.cli import main
+
+
 # An example parse tree looks like:
 
 #                            nonterminal A