Fixed rubocop failures