Fix rubocop complaints