Rubocop fix