return comp_id;
        }
 
-       ret = mtk_ddp_comp_init(dev, dev->of_node, &priv->ddp_comp, comp_id,
+       ret = mtk_ddp_comp_init(dev->of_node, &priv->ddp_comp, comp_id,
                                &mtk_disp_color_funcs);
        if (ret) {
                if (ret != -EPROBE_DEFER)
 
                return comp_id;
        }
 
-       ret = mtk_ddp_comp_init(dev, dev->of_node, &priv->ddp_comp, comp_id,
+       ret = mtk_ddp_comp_init(dev->of_node, &priv->ddp_comp, comp_id,
                                &mtk_disp_ovl_funcs);
        if (ret) {
                if (ret != -EPROBE_DEFER)
 
                return comp_id;
        }
 
-       ret = mtk_ddp_comp_init(dev, dev->of_node, &priv->ddp_comp, comp_id,
+       ret = mtk_ddp_comp_init(dev->of_node, &priv->ddp_comp, comp_id,
                                &mtk_disp_rdma_funcs);
        if (ret) {
                if (ret != -EPROBE_DEFER)
 
                return comp_id;
        }
 
-       ret = mtk_ddp_comp_init(dev, dev->of_node, &dpi->ddp_comp, comp_id,
+       ret = mtk_ddp_comp_init(dev->of_node, &dpi->ddp_comp, comp_id,
                                &mtk_dpi_funcs);
        if (ret) {
                dev_err(dev, "Failed to initialize component: %d\n", ret);
 
        return ret;
 }
 
-int mtk_ddp_comp_init(struct device *dev, struct device_node *node,
-                     struct mtk_ddp_comp *comp, enum mtk_ddp_comp_id comp_id,
-                     const struct mtk_ddp_comp_funcs *funcs)
+int mtk_ddp_comp_init(struct device_node *node, struct mtk_ddp_comp *comp,
+                     enum mtk_ddp_comp_id comp_id, const struct mtk_ddp_comp_funcs *funcs)
 {
+       struct platform_device *comp_pdev;
+       struct device *dev;
        enum mtk_ddp_comp_type type;
        struct device_node *larb_node;
        struct platform_device *larb_pdev;
        if (IS_ERR(comp->clk))
                return PTR_ERR(comp->clk);
 
+       comp_pdev = of_find_device_by_node(node);
+       if (!comp_pdev) {
+               DRM_INFO("Waiting for device %s\n", node->full_name);
+               return -EPROBE_DEFER;
+       }
+       dev = &comp_pdev->dev;
+
 #if IS_REACHABLE(CONFIG_MTK_CMDQ)
        if (of_address_to_resource(node, 0, &res) != 0) {
                dev_err(dev, "Missing reg in %s node\n", node->full_name);
-               put_device(&larb_pdev->dev);
                return -EINVAL;
        }
        comp->regs_pa = res.start;
 
                        enum mtk_ddp_comp_type comp_type);
 unsigned int mtk_drm_find_possible_crtc_by_comp(struct drm_device *drm,
                                                struct mtk_ddp_comp ddp_comp);
-int mtk_ddp_comp_init(struct device *dev, struct device_node *comp_node,
-                     struct mtk_ddp_comp *comp, enum mtk_ddp_comp_id comp_id,
-                     const struct mtk_ddp_comp_funcs *funcs);
+int mtk_ddp_comp_init(struct device_node *comp_node, struct mtk_ddp_comp *comp,
+                     enum mtk_ddp_comp_id comp_id, const struct mtk_ddp_comp_funcs *funcs);
 int mtk_ddp_comp_register(struct drm_device *drm, struct mtk_ddp_comp *comp);
 void mtk_ddp_comp_unregister(struct drm_device *drm, struct mtk_ddp_comp *comp);
 void mtk_dither_set(struct mtk_ddp_comp *comp, unsigned int bpc,
 
                                goto err_node;
                        }
 
-                       ret = mtk_ddp_comp_init(dev->parent, node, comp,
+                       ret = mtk_ddp_comp_init(node, comp,
                                                comp_id, NULL);
                        if (ret) {
                                of_node_put(node);
 
                goto err_unregister_host;
        }
 
-       ret = mtk_ddp_comp_init(dev, dev->of_node, &dsi->ddp_comp, comp_id,
+       ret = mtk_ddp_comp_init(dev->of_node, &dsi->ddp_comp, comp_id,
                                &mtk_dsi_funcs);
        if (ret) {
                dev_err(dev, "Failed to initialize component: %d\n", ret);