diff --git a/project/dv/apps/components/AiGrid.vue b/project/dv/apps/components/AiGrid.vue
index ca28b1da..1c774a6a 100644
--- a/project/dv/apps/components/AiGrid.vue
+++ b/project/dv/apps/components/AiGrid.vue
@@ -5,7 +5,7 @@
         ref="tree"
         id="tree"
         :style="{left: x, top: y, transform: `scale(${scale}) translate(-50%, -50%) `, 'transform-origin': `${0} ${0}`}">
-      <VueOkrTree
+      <ai-okr-tree
           :props="props"
           node-key="id"
           show-collapsable
@@ -15,23 +15,16 @@
           ref="VueOkrTree"
           @node-click="onNodeClick"
           :data="treeData">
-      </VueOkrTree>
+      </ai-okr-tree>
     </div>
   </div>
 </template>
 
 <script>
-import AiOkrTree from "dvcp-dv-ui/components/AiOkrTree/AiOkrTree"
-
 export default {
   name: 'AiGrid',
-
   props: ['instance'],
 
-  components: {
-    VueOkrTree: AiOkrTree
-  },
-
   data() {
     return {
       scale: 1,