diff --git a/generate-flag-types b/generate-flag-types
index 6244ff9..aa51154 100755
--- a/generate-flag-types
+++ b/generate-flag-types
@@ -193,6 +193,8 @@ def _write_altsrc_flag_types(outfile, types):
     _fwrite(outfile, """\
         package altsrc
 
+        import "gopkg.in/urfave/cli.v2"
+
         // WARNING: This file is generated!
 
         """)