diff --git a/.travis.yml b/.travis.yml
index 75d9ea3..ad53098 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -13,6 +13,7 @@ cache:
     - node_modules
 
 before_script:
+  - go get github.com/urfave/cli/flag-gen || true
   - go get github.com/urfave/gfmrun/... || true
   - go get golang.org/x/tools/cmd/goimports
   - if [ ! -f node_modules/.bin/markdown-toc ] ; then
@@ -20,7 +21,7 @@ before_script:
     fi
 
 script:
-  - ./runtests gen
+  - go generate ./...
   - ./runtests vet
   - ./runtests test
   - ./runtests gfmrun