diff --git a/src/controller/dashboard.controller.js b/src/controller/dashboard.controller.js
index 545657f..54026de 100644
--- a/src/controller/dashboard.controller.js
+++ b/src/controller/dashboard.controller.js
@@ -356,7 +356,18 @@ module.exports = {
       const { id, type } = req.body;
       if (!id || !type) return res.send({ success: false, message: "Missing id or type" });
 
-      const query = {};
+      let query = {};
+
+      switch (type) {
+        case "level":
+        case "sublevel":
+          query = { $or: [{ main_level_id: id, parent_sublevel_id: id }] };
+          break;
+
+        default:
+          query[`${type}_id`] = id;
+          break;
+      }
       query[`${type}_id`] = id;
 
       const childrenData = await getChildModel(type).find(query);
diff --git a/src/controller/sublevel.controller.js b/src/controller/sublevel.controller.js
index 4c353a8..28e4846 100644
--- a/src/controller/sublevel.controller.js
+++ b/src/controller/sublevel.controller.js
@@ -31,6 +31,7 @@ module.exports = {
    * Create a sublevel
    */
   createSubLevel: async (req, res, next) => {
+    console.log(req.body);
     const { name, type, specs, parent_id, parentIsLevel, positions } = req.body;
 
     if (!(name && type && parent_id && positions)) {
@@ -70,7 +71,7 @@ module.exports = {
         res.status(404);
         return;
       }
-      req.send(sublevelData);
+      res.send(sublevelData);
     } catch (error) {
       next(error);
     }
diff --git a/src/controller/utils/sublevel.js b/src/controller/utils/sublevel.js
index 3e63a93..ce6cf75 100644
--- a/src/controller/utils/sublevel.js
+++ b/src/controller/utils/sublevel.js
@@ -58,7 +58,7 @@ const deleteSubLevelTreeFromRoot = async (root_sub_level_id) => {
 const addSublevelToParent = async (payload, parent_id, parentIsLevel) => {
   if (parentIsLevel) {
     // add sublevel to parent
-    const parentData = await Sublevel.findById(parent_id);
+    const parentData = await Level.findById(parent_id);
     parentData.sub_levels.push(payload);
     return await parentData.save();
   } else {