diff --git a/htfft/stage_gen.py b/htfft/stage_gen.py
index fa8aa6a..d79d8a7 100644
--- a/htfft/stage_gen.py
+++ b/htfft/stage_gen.py
@@ -3,8 +3,7 @@
 import jinja2
 from fusesoc.capi2.generator import Generator
 
-from htfft import helper, conversions
-import htfft_gen
+from htfft import helper, conversions, htfft_gen
 
 basedir = os.path.abspath(os.path.dirname(__file__))