Fixes Go lint errors

Moving it to the command (instead of package) is an interesting choice, though. I think I could go either way.