Fix code formatted by gofmt