Fixing rubocop warning