Update to latest go/types API