Fixing rubocop offence