diff --git a/packages/behavior/src/traits/behaved.js b/packages/behavior/src/traits/behaved.js
index 9fadf1d..49c3820 100644
--- a/packages/behavior/src/traits/behaved.js
+++ b/packages/behavior/src/traits/behaved.js
@@ -37,15 +37,6 @@ export default (latus) => class Behaved extends decorate(Trait) {
     } = this.constructor.defaultState());
   }
 
-  static behaviorTypes() {
-    return {
-      context: {
-        type: 'context',
-        label: 'Context',
-      },
-    };
-  }
-
   get context() {
     return this.#context;
   }
@@ -77,6 +68,10 @@ export default (latus) => class Behaved extends decorate(Trait) {
 
   static children(entity) {
     return {
+      context: {
+        type: 'context',
+        label: 'Context',
+      },
       isBehaving: {
         type: 'bool',
         label: 'Is behaving',