1 files changed, 13 insertions, 1 deletions
diff --git a/src/core/model/Domain.cpp b/src/core/model/Domain.cpp
index 360aa83..ef505dd 100644
--- a/src/core/model/Domain.cpp
+++ b/src/core/model/Domain.cpp
@@ -469,6 +469,18 @@ bool Domain::doValidate(Logger &logger) const
 	       continueValidationCheckDuplicates(typesystems, logger);
 }
 
+void Domain::doReference(Handle<Node> node)
+{
+	if (node->isa(RttiTypes::Domain)) {
+		referenceTypesystem(node.cast<Typesystem>());
+	}
+}
+
+RttiSet Domain::doGetReferenceTypes() const
+{
+	return RttiSet{&RttiTypes::Domain};
+}
+
 void Domain::addStructuredClass(Handle<StructuredClass> s)
 {
 	// only add it if we need to.
@@ -559,7 +571,7 @@ const Rtti StructuredClass =
 const Rtti AnnotationClass =
     RttiBuilder<ousia::AnnotationClass>("AnnotationClass").parent(&Descriptor);
 const Rtti Domain = RttiBuilder<ousia::Domain>("Domain")
-                        .parent(&Node)
+                        .parent(&RootNode)
                         .composedOf({&StructuredClass, &AnnotationClass});
 }
 }