diff --git a/tmp.py b/tmp.py
index 7f7b969..fb6bacf 100644
--- a/tmp.py
+++ b/tmp.py
@@ -1,5 +1,7 @@
 from pathlib import Path
 
+from .pyflybygen import pyflybygen
+
 statements = """
 from __future__ import (absolute_import,
                         division)