From 49b8f5f39fd03639bdd52f3e834ef0b536835950 Mon Sep 17 00:00:00 2001
From: Janice Collins <jcollins@google.com>
Date: Mon, 16 Aug 2021 13:12:57 -0700
Subject: [PATCH] Switch to using CompilationUnitElement.classes

---
 lib/src/model/library.dart | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/src/model/library.dart b/lib/src/model/library.dart
index 0d473879b0..b79109b623 100644
--- a/lib/src/model/library.dart
+++ b/lib/src/model/library.dart
@@ -102,13 +102,13 @@ class Library extends ModelElement with Categorization, TopLevelContainer {
       CompilationUnitElement compilationUnit) {
     return quiver.concat([
       compilationUnit.accessors,
+      compilationUnit.classes,
       compilationUnit.enums,
       compilationUnit.extensions,
       compilationUnit.functions,
       compilationUnit.mixins,
       compilationUnit.topLevelVariables,
       compilationUnit.typeAliases,
-      compilationUnit.types,
     ]);
   }