Fix some rubocop issues