Fixing for rubocop