diff --git a/typeutil.go b/typeutil.go
index 3a68dc8..7a701a3 100644
--- a/typeutil.go
+++ b/typeutil.go
@@ -11,6 +11,7 @@ import (
 	"io"
 	"sort"
 	"strconv"
+	"strings"
 )
 
 // walkNamedTypes runs the callback for all named types contained in the given type.
@@ -168,10 +169,14 @@ func newFileScope(imp types.Importer, pkg *types.Package) *fileScope {
 func (s *fileScope) writeImportDecl(w io.Writer) {
 	fmt.Fprintln(w, "import (")
 	for _, pkg := range s.imports {
+		path := pkg.Path()
+		if idx := strings.LastIndex(path, "/vendor/"); idx >= 0 {
+			path = path[idx+8:]
+		}
 		if s.importNames[pkg.Path()] != pkg.Name() {
-			fmt.Fprintf(w, "\t%s %q\n", s.importNames[pkg.Path()], pkg.Path())
+			fmt.Fprintf(w, "\t%s %q\n", s.importNames[pkg.Path()], path)
 		} else {
-			fmt.Fprintf(w, "\t%q\n", pkg.Path())
+			fmt.Fprintf(w, "\t%q\n", path)
 		}
 	}
 	fmt.Fprintln(w, ")")