Fix TS bundle

src/trait_manager/index.ts

@@ -20,7 +20,7 @@ import TraitNumberView from './view/TraitNumberView';
 import TraitSelectView from './view/TraitSelectView';
 import TraitView from './view/TraitView';
 import TraitsView from './view/TraitsView';
-import { getItemsByCategory } from '../abstract/ModuleCategory';
+import Category, { getItemsByCategory } from '../abstract/ModuleCategory';
 import Trait from './model/Trait';
 
 export default class TraitManager extends Module<TraitManagerConfigModule> {
@@ -138,7 +138,7 @@ export default class TraitManager extends Module<TraitManagerConfigModule> {
    * Get trait categories from the currently selected component.
    * @return {Array<Category>}
    */
-  getCategories() {
+  getCategories(): Category[] {
     const cmp = this.state.get('component');
     const categories = cmp?.traits.categories?.models || [];
     return [...categories];

src/trait_manager/model/Traits.ts

@@ -6,6 +6,7 @@ import EditorModel from '../../editor/model/Editor';
 import Trait, { TraitProperties } from './Trait';
 import TraitFactory from './TraitFactory';
 import Categories from '../../abstract/ModuleCategories';
+import TraitManager from '..';
 
 export default class Traits extends CollectionWithCategories<Trait> {
   em: EditorModel;
@@ -23,7 +24,7 @@ export default class Traits extends CollectionWithCategories<Trait> {
     this.tf = new TraitFactory(tmOpts);
   }
 
-  get module() {
+  get module(): TraitManager {
     return this.em.Traits;
   }