Remove compiler error

I got an error "function ends without a return statement" on go1.0.2