mutex_unlock(&dpu_enc->enc_lock);
 }
 
-static enum dpu_intf dpu_encoder_get_intf(const struct dpu_mdss_cfg *catalog,
+static struct dpu_hw_intf *dpu_encoder_get_intf(const struct dpu_mdss_cfg *catalog,
+               struct dpu_rm *dpu_rm,
                enum dpu_intf_type type, u32 controller_id)
 {
        int i = 0;
 
        if (type == INTF_WB)
-               return INTF_MAX;
+               return NULL;
 
        for (i = 0; i < catalog->intf_count; i++) {
                if (catalog->intf[i].type == type
                    && catalog->intf[i].controller_id == controller_id) {
-                       return catalog->intf[i].id;
+                       return dpu_rm_get_intf(dpu_rm, catalog->intf[i].id);
                }
        }
 
-       return INTF_MAX;
+       return NULL;
 }
 
 void dpu_encoder_vblank_callback(struct drm_encoder *drm_enc,
                 * h_tile_instance_ids[2] = {1, 0}; DSI1 = left, DSI0 = right
                 */
                u32 controller_id = disp_info->h_tile_instance[i];
-               enum dpu_intf intf_idx;
 
                if (disp_info->num_of_h_tiles > 1) {
                        if (i == 0)
                DPU_DEBUG("h_tile_instance %d = %d, split_role %d\n",
                                i, controller_id, phys_params.split_role);
 
-               intf_idx = dpu_encoder_get_intf(dpu_kms->catalog,
-                                                           disp_info->intf_type,
-                                                           controller_id);
-
-               if (intf_idx >= INTF_0 && intf_idx < INTF_MAX)
-                       phys_params.hw_intf = dpu_rm_get_intf(&dpu_kms->rm, intf_idx);
+               phys_params.hw_intf = dpu_encoder_get_intf(dpu_kms->catalog, &dpu_kms->rm,
+                                                          disp_info->intf_type,
+                                                          controller_id);
 
                if (disp_info->intf_type == INTF_WB && controller_id < WB_MAX)
                        phys_params.hw_wb = dpu_rm_get_wb(&dpu_kms->rm, controller_id);
                        DPU_ERROR_ENC(dpu_enc, "failed to add phys encs\n");
                        break;
                }
-
        }
 
        mutex_unlock(&dpu_enc->enc_lock);