diff --git a/py_gen/templates/toplevel_init.py b/py_gen/templates/toplevel_init.py
index 3151404..72f4c81 100644
--- a/py_gen/templates/toplevel_init.py
+++ b/py_gen/templates/toplevel_init.py
@@ -25,10 +25,13 @@
 :: # EPL for the specific language governing permissions and limitations
 :: # under the EPL.
 ::
+:: import of_g
 :: include('_copyright.py')
 
 :: include('_autogen.py')
 
+version_names = ${repr(of_g.param_version_names)}
+
 def protocol(ver):
     """
     Import and return the protocol module for the given wire version.