diff --git a/py_gen/codegen.py b/py_gen/codegen.py
index 70053f7..3c98333 100644
--- a/py_gen/codegen.py
+++ b/py_gen/codegen.py
@@ -29,6 +29,7 @@
 import struct
 import of_g
 import loxi_front_end.type_maps as type_maps
+import template_utils
 import loxi_utils.loxi_utils as utils
 import util
 import oftype