spin_lock(&mdev->lock);
 
        media_device_for_each_entity(entity, mdev) {
-               if ((entity->id == id && !next) ||
-                   (entity->id > id && next)) {
+               if (((media_entity_id(entity) == id) && !next) ||
+                   ((media_entity_id(entity) > id) && next)) {
                        spin_unlock(&mdev->lock);
                        return entity;
                }
        if (ent == NULL)
                return -EINVAL;
 
-       u_ent.id = ent->id;
+       u_ent.id = media_entity_id(ent);
        if (ent->name)
                strlcpy(u_ent.name, ent->name, sizeof(u_ent.name));
        u_ent.type = ent->type;
 static void media_device_kpad_to_upad(const struct media_pad *kpad,
                                      struct media_pad_desc *upad)
 {
-       upad->entity = kpad->entity->id;
+       upad->entity = media_entity_id(kpad->entity);
        upad->index = kpad->index;
        upad->flags = kpad->flags;
 }
 
        graph->stack[graph->top].entity = NULL;
        bitmap_zero(graph->entities, MEDIA_ENTITY_ENUM_MAX_ID);
 
-       if (WARN_ON(entity->id >= MEDIA_ENTITY_ENUM_MAX_ID))
+       if (WARN_ON(media_entity_id(entity) >= MEDIA_ENTITY_ENUM_MAX_ID))
                return;
 
-       __set_bit(entity->id, graph->entities);
+       __set_bit(media_entity_id(entity), graph->entities);
        stack_push(graph, entity);
 }
 EXPORT_SYMBOL_GPL(media_entity_graph_walk_start);
 
                /* Get the entity in the other end of the link . */
                next = media_entity_other(entity, link);
-               if (WARN_ON(next->id >= MEDIA_ENTITY_ENUM_MAX_ID))
+               if (WARN_ON(media_entity_id(next) >= MEDIA_ENTITY_ENUM_MAX_ID))
                        return NULL;
 
                /* Has the entity already been visited? */
-               if (__test_and_set_bit(next->id, graph->entities)) {
+               if (__test_and_set_bit(media_entity_id(next), graph->entities)) {
                        link_top(graph)++;
                        continue;
                }
 
                        break;
 
                /* Ensure the branch has no loop. */
-               if (entities & (1 << entity->subdev.entity.id))
+               if (entities & (1 << media_entity_id(&entity->subdev.entity)))
                        return -EPIPE;
 
-               entities |= 1 << entity->subdev.entity.id;
+               entities |= 1 << media_entity_id(&entity->subdev.entity);
 
                /* UDS can't be chained. */
                if (entity->type == VSP1_ENTITY_UDS) {
 
        return entity->type & MEDIA_ENT_SUBTYPE_MASK;
 }
 
+static inline u32 media_entity_id(struct media_entity *entity)
+{
+       return entity->id;
+}
+
 #define MEDIA_ENTITY_ENUM_MAX_DEPTH    16
 #define MEDIA_ENTITY_ENUM_MAX_ID       64