Clean up from go fmt