Fixing rubocop issue