1 files changed, 7 insertions, 4 deletions
diff --git a/examples/uml/profile-definition/main.cpp b/examples/uml/profile-definition/main.cpp
index 3ef20ca6..b359c7eb 100644
--- a/examples/uml/profile-definition/main.cpp
+++ b/examples/uml/profile-definition/main.cpp
@@ -79,11 +79,18 @@ int main ()
     // Adding attribute to stereotype
     QUmlPrimitiveType *booleanPrimitiveType = new QUmlPrimitiveType;
     booleanPrimitiveType->setName("boolean");
+
     QUmlProperty *property = new QUmlProperty;
     property->setName("isTransient");
     property->setType(booleanPrimitiveType);
     stereotype->addOwnedAttribute(property);
 
+    // Adding stereotype attribute that play the role of extension end
+    QUmlProperty *stereotypeProperty = new QUmlProperty;
+    stereotypeProperty->setName("base_class");
+    stereotypeProperty->setType(class_);
+    stereotype->addOwnedAttribute(stereotypeProperty);
+
     // Create UML meta-model element import
     QUmlElementImport *elementImport = new QUmlElementImport;
     elementImport->setImportedElement(umlModel->packagedElements().toList().first());
@@ -107,10 +114,6 @@ int main ()
     QUmlExtension *extension = new QUmlExtension;
     extension->setName("class_stereotype");
 
-    QUmlProperty *stereotypeProperty = new QUmlProperty;
-    stereotypeProperty->setName("base_class");
-    stereotypeProperty->setType(class_);
-
     QUmlExtensionEnd *extensionEnd = new QUmlExtensionEnd;
     extensionEnd->setName("extension_stereotype");
     extensionEnd->setType(stereotype);