Updated to work with vim-go.