Fixed missing return statement that broke the build in go 1.0