1 files changed, 5 insertions, 1 deletions
diff --git a/src/libs/modelinglib/qmt/style/styledrelation.h b/src/libs/modelinglib/qmt/style/styledrelation.h
index 422947908e..27e8735463 100644
--- a/src/libs/modelinglib/qmt/style/styledrelation.h
+++ b/src/libs/modelinglib/qmt/style/styledrelation.h
@@ -9,21 +9,25 @@ namespace qmt {
 
 class DRelation;
 class DObject;
+class CustomRelation;
 
 class QMT_EXPORT StyledRelation
 {
 public:
-    StyledRelation(const DRelation *relation, const DObject *endA, const DObject *endB);
+    StyledRelation(const DRelation *relation, const DObject *endA, const DObject *endB,
+                   const CustomRelation *customRelation);
     ~StyledRelation();
 
     const DRelation *relation() const { return m_relation; }
     const DObject *endA() const { return m_endA; }
     const DObject *endB() const { return m_endB; }
+    const CustomRelation *customRelation() const { return m_customRelation; }
 
 private:
     const DRelation *m_relation = nullptr;
     const DObject *m_endA = nullptr;
     const DObject *m_endB = nullptr;
+    const CustomRelation *m_customRelation = nullptr;
 };
 
 } // namespace qmt