diff --git a/cdp/__init__.py b/cdp/__init__.py
index dbb969c..e29a12e 100644
--- a/cdp/__init__.py
+++ b/cdp/__init__.py
@@ -4,6 +4,7 @@
 # changes, edit the generator and regenerate all of the modules.
 
 import cdp.util
+import cdp.connection
 
 import cdp.accessibility
 import cdp.animation
diff --git a/generator/generate.py b/generator/generate.py
index a386654..8925859 100644
--- a/generator/generate.py
+++ b/generator/generate.py
@@ -964,7 +964,8 @@ def generate_init(init_path, domains):
     '''
     with init_path.open('w') as init_file:
         init_file.write(INIT_HEADER)
-        init_file.write('import cdp.util\n\n')
+        init_file.write('import cdp.util\n')
+        init_file.write('import cdp.connection\n\n')
         for domain in domains:
             init_file.write('import cdp.{}\n'.format(domain.module))