diff --git a/generator.py b/generator.py
index 74150b7..66c53d1 100755
--- a/generator.py
+++ b/generator.py
@@ -1828,6 +1828,8 @@ def qemuBuildWrappers(module):
     fd.write("# Automatically written part of python bindings for libvirt\n")
     fd.write("#\n")
 
+    fd.write("import sys\n")
+
     fd.write("try:\n")
     fd.write("    import libvirtmod_qemu\n")
     fd.write("except ImportError:\n")
@@ -1951,6 +1953,8 @@ def lxcBuildWrappers(module):
     if extra is not None:
         extra.close()
 
+    fd.write("import sys\n")
+
     fd.write("try:\n")
     fd.write("    import libvirtmod_lxc\n")
     fd.write("except ImportError:\n")