gofmt: fix for gofmt rewrite feature

Fixes #643.

R=rsc
CC=golang-dev
https://golang.org/cl/576041
2 files changed