diff --git a/packages/core/src/ExtensionManager.ts b/packages/core/src/ExtensionManager.ts
index a32d023f..178beaa6 100644
--- a/packages/core/src/ExtensionManager.ts
+++ b/packages/core/src/ExtensionManager.ts
@@ -53,38 +53,16 @@ export default class ExtensionManager {
     })
   }
 
-  // get topNode() {
-  //   return getTopNodeFromExtensions(this.extensions)
-  // }
-
-  // get nodes(): any {
-  //   return getNodesFromExtensions(this.extensions)
-  // }
-  
-  // get marks(): any {
-  //   return getMarksFromExtensions(this.extensions)
-  // }
-
   get topNode(): any {
-    const topNode = collect(this.extensions).firstWhere('config.topNode', true)
-
-    if (topNode) {
-      return topNode.config.name
-    }
+    return getTopNodeFromExtensions(this.extensions)
   }
 
   get nodes(): any {
-    return collect(this.extensions)
-      .where('type', 'node')
-      .mapWithKeys((extension: Node) => [extension.config.name, extension.config.schema])
-      .all()
+    return getNodesFromExtensions(this.extensions)
   }
-
+  
   get marks(): any {
-    return collect(this.extensions)
-      .where('type', 'mark')
-      .mapWithKeys((extension: Mark) => [extension.config.name, extension.config.schema])
-      .all()
+    return getMarksFromExtensions(this.extensions)
   }
 
   get plugins(): Plugin[] {
diff --git a/packages/core/src/utils/getMarksFromExtensions.ts b/packages/core/src/utils/getMarksFromExtensions.ts
index fca51100..63c890bf 100644
--- a/packages/core/src/utils/getMarksFromExtensions.ts
+++ b/packages/core/src/utils/getMarksFromExtensions.ts
@@ -4,7 +4,7 @@ import { Extensions } from '../types'
 
 export default function getMarksFromExtensions(extensions: Extensions): any {
   return collect(extensions)
-    .where('extensionType', 'mark')
-    .mapWithKeys((extension: Mark) => [extension.name, extension.schema()])
+    .where('type', 'mark')
+    .mapWithKeys((extension: Mark) => [extension.config.name, extension.config.schema])
     .all()
 }
\ No newline at end of file
diff --git a/packages/core/src/utils/getNodesFromExtensions.ts b/packages/core/src/utils/getNodesFromExtensions.ts
index 94604110..409c04fc 100644
--- a/packages/core/src/utils/getNodesFromExtensions.ts
+++ b/packages/core/src/utils/getNodesFromExtensions.ts
@@ -4,7 +4,7 @@ import { Extensions } from '../types'
 
 export default function getNodesFromExtensions(extensions: Extensions): any {
   return collect(extensions)
-    .where('extensionType', 'node')
-    .mapWithKeys((extension: Node) => [extension.name, extension.schema()])
+    .where('type', 'node')
+    .mapWithKeys((extension: Node) => [extension.config.name, extension.config.schema])
     .all()
 }
\ No newline at end of file
diff --git a/packages/core/src/utils/getTopNodeFromExtensions.ts b/packages/core/src/utils/getTopNodeFromExtensions.ts
index 360bc57e..a7ce3e63 100644
--- a/packages/core/src/utils/getTopNodeFromExtensions.ts
+++ b/packages/core/src/utils/getTopNodeFromExtensions.ts
@@ -2,9 +2,9 @@ import collect from 'collect.js'
 import { Extensions } from '../types'
 
 export default function getTopNodeFromExtensions(extensions: Extensions): any {
-  const topNode = collect(extensions).firstWhere('topNode', true)
+  const topNode = collect(extensions).firstWhere('config.topNode', true)
 
   if (topNode) {
-    return topNode.name
+    return topNode.config.name
   }
 }
\ No newline at end of file