Rubocop fixing