diff --git a/src/parser/ast.yaml b/src/parser/ast.yaml
index fd982ad..694b428 100644
--- a/src/parser/ast.yaml
+++ b/src/parser/ast.yaml
@@ -1,3 +1,4 @@
+# $schema: ./ast.schema.yaml
 CompilationUnit:
   children:
     - namespace
@@ -29,6 +30,8 @@ ModuleLevelDeclaration:
       build: ModuleDeclaration
     - rule: Interface
       build: InterfaceDeclaration
+    - rule: Class
+      build: ClassDeclaration
     - FunctionDefinition
     - PlatformFunction
 ModuleDeclaration:
@@ -46,7 +49,7 @@ ModuleDeclaration:
     - declarations:
         rule: ModuleLevelDeclaration
         vec: true
-Class:
+ClassDeclaration:
   children:
     - is_public:
         rule: Pub