"au8522", 0x8e >> 1, NULL);
                if (sd == NULL)
                        pr_err("analog subdev registration failed\n");
+#ifdef CONFIG_MEDIA_CONTROLLER
+               if (sd)
+                       dev->decoder = &sd->entity;
+#endif
        }
 
        /* Setup tuners */
 
 {
 #ifdef CONFIG_MEDIA_CONTROLLER
        struct media_device *mdev = dev->media_dev;
-       struct media_entity  *entity, *decoder = NULL, *source;
+       struct media_entity  *entity, *source;
        struct media_link *link, *found_link = NULL;
        int i, ret, active_links = 0;
 
-       if (!mdev)
+       if (!mdev || !dev->decoder)
                return 0;
 
        /*
         * do DVB streaming while the DMA engine is being used for V4L2,
         * this should be enough for the actual needs.
         */
-       media_device_for_each_entity(entity, mdev) {
-               if (entity->type == MEDIA_ENT_T_V4L2_SUBDEV_DECODER) {
-                       decoder = entity;
-                       break;
-               }
-       }
-       if (!decoder)
-               return 0;
-
-       for (i = 0; i < decoder->num_links; i++) {
-               link = &decoder->links[i];
-               if (link->sink->entity == decoder) {
+       for (i = 0; i < dev->decoder->num_links; i++) {
+               link = &dev->decoder->links[i];
+               if (link->sink->entity == dev->decoder) {
                        found_link = link;
                        if (link->flags & MEDIA_LNK_FL_ENABLED)
                                active_links++;
 
 #ifdef CONFIG_MEDIA_CONTROLLER
        struct media_device *media_dev;
        struct media_pad video_pad, vbi_pad;
+       struct media_entity *decoder;
 #endif
 };