*
         * Link HDMI->TXA, and AFE->TXB directly.
         */
-       if (is_txa(tx)) {
+       if (is_txa(tx) && is_hdmi_enabled(state))
                return adv748x_csi2_register_link(tx, sd->v4l2_dev,
                                                  &state->hdmi.sd,
                                                  ADV748X_HDMI_SOURCE);
-       } else {
+       if (!is_txa(tx) && is_afe_enabled(state))
                return adv748x_csi2_register_link(tx, sd->v4l2_dev,
                                                  &state->afe.sd,
                                                  ADV748X_AFE_SOURCE);
-       }
+       return 0;
 }
 
 static const struct v4l2_subdev_internal_ops adv748x_csi2_internal_ops = {
 
 #define adv748x_sd_to_csi2(sd) container_of(sd, struct adv748x_csi2, sd)
 #define is_tx_enabled(_tx) ((_tx)->state->endpoints[(_tx)->port] != NULL)
 #define is_txa(_tx) ((_tx) == &(_tx)->state->txa)
+#define is_afe_enabled(_state)                                 \
+       ((_state)->endpoints[ADV748X_PORT_AIN0] != NULL ||      \
+        (_state)->endpoints[ADV748X_PORT_AIN1] != NULL ||      \
+        (_state)->endpoints[ADV748X_PORT_AIN2] != NULL ||      \
+        (_state)->endpoints[ADV748X_PORT_AIN3] != NULL ||      \
+        (_state)->endpoints[ADV748X_PORT_AIN4] != NULL ||      \
+        (_state)->endpoints[ADV748X_PORT_AIN5] != NULL ||      \
+        (_state)->endpoints[ADV748X_PORT_AIN6] != NULL ||      \
+        (_state)->endpoints[ADV748X_PORT_AIN7] != NULL)
+#define is_hdmi_enabled(_state) ((_state)->endpoints[ADV748X_PORT_HDMI] != NULL)
 
 enum adv748x_hdmi_pads {
        ADV748X_HDMI_SINK,