From 65b3a050d9f278088449d1961dea73f7f26da014 Mon Sep 17 00:00:00 2001
From: Artur Arseniev <artur.catch@hotmail.it>
Date: Sat, 20 Jan 2024 14:35:50 +0400
Subject: [PATCH] Up traits

---
 src/trait_manager/model/Traits.ts | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/trait_manager/model/Traits.ts b/src/trait_manager/model/Traits.ts
index 302e474a1..8d26b6c76 100644
--- a/src/trait_manager/model/Traits.ts
+++ b/src/trait_manager/model/Traits.ts
@@ -13,13 +13,17 @@ export default class Traits extends Collection<Trait> {
   constructor(coll: TraitProperties[], options: { em: EditorModel }) {
     super(coll);
     this.em = options.em;
-    this.listenTo(this, 'add', this.handleAdd);
-    this.listenTo(this, 'reset', this.handleReset);
-    const tm = this.em?.Traits;
+    this.on('add', this.handleAdd);
+    this.on('reset', this.handleReset);
+    const tm = this.module;
     const tmOpts = tm?.getConfig();
     this.tf = new TraitFactory(tmOpts);
   }
 
+  get module() {
+    return this.em.Traits;
+  }
+
   handleReset(coll: TraitProperties[], { previousModels = [] }: { previousModels?: Trait[] } = {}) {
     previousModels.forEach(model => model.trigger('remove'));
   }