Fix linting issues after golang update

Signed-off-by: Henry Wang <henrwang@nvidia.com>