con->cable = NULL;
 }
 
+static int ucsi_check_connector_capability(struct ucsi_connector *con)
+{
+       u64 command;
+       int ret;
+
+       if (!con->partner || con->ucsi->version < UCSI_VERSION_2_1)
+               return 0;
+
+       command = UCSI_GET_CONNECTOR_CAPABILITY | UCSI_CONNECTOR_NUMBER(con->num);
+       ret = ucsi_send_command(con->ucsi, command, &con->cap, sizeof(con->cap));
+       if (ret < 0) {
+               dev_err(con->ucsi->dev, "GET_CONNECTOR_CAPABILITY failed (%d)\n", ret);
+               return ret;
+       }
+
+       typec_partner_set_pd_revision(con->partner,
+               UCSI_CONCAP_FLAG_PARTNER_PD_MAJOR_REV_AS_BCD(con->cap.flags));
+
+       return ret;
+}
+
 static void ucsi_pwr_opmode_change(struct ucsi_connector *con)
 {
        switch (UCSI_CONSTAT_PWR_OPMODE(con->status.flags)) {
                ucsi_partner_task(con, ucsi_get_src_pdos, 30, 0);
                ucsi_partner_task(con, ucsi_check_altmodes, 30, HZ);
                ucsi_partner_task(con, ucsi_register_partner_pdos, 1, HZ);
+               ucsi_partner_task(con, ucsi_check_connector_capability, 1, HZ);
                break;
        case UCSI_CONSTAT_PWR_OPMODE_TYPEC1_5:
                con->rdo = 0;
 
        desc.identity = &con->partner_identity;
        desc.usb_pd = pwr_opmode == UCSI_CONSTAT_PWR_OPMODE_PD;
-       desc.pd_revision = UCSI_CONCAP_FLAG_PARTNER_PD_MAJOR_REV_AS_BCD(con->cap.flags);
 
        partner = typec_register_partner(con->port, &desc);
        if (IS_ERR(partner)) {
                        con->num, u_role);
 }
 
-static int ucsi_check_connector_capability(struct ucsi_connector *con)
-{
-       u64 command;
-       int ret;
-
-       if (!con->partner || con->ucsi->version < UCSI_VERSION_2_0)
-               return 0;
-
-       command = UCSI_GET_CONNECTOR_CAPABILITY | UCSI_CONNECTOR_NUMBER(con->num);
-       ret = ucsi_send_command(con->ucsi, command, &con->cap, sizeof(con->cap));
-       if (ret < 0) {
-               dev_err(con->ucsi->dev, "GET_CONNECTOR_CAPABILITY failed (%d)\n", ret);
-               return ret;
-       }
-
-       typec_partner_set_pd_revision(con->partner,
-               UCSI_CONCAP_FLAG_PARTNER_PD_MAJOR_REV_AS_BCD(con->cap.flags));
-
-       return ret;
-}
-
 static int ucsi_check_connection(struct ucsi_connector *con)
 {
        u8 prev_flags = con->status.flags;
                if (con->status.flags & UCSI_CONSTAT_CONNECTED) {
                        ucsi_register_partner(con);
                        ucsi_partner_task(con, ucsi_check_connection, 1, HZ);
-                       ucsi_partner_task(con, ucsi_check_connector_capability, 1, HZ);
                        if (con->ucsi->cap.features & UCSI_CAP_GET_PD_MESSAGE)
                                ucsi_partner_task(con, ucsi_get_partner_identity, 1, HZ);
                        if (con->ucsi->cap.features & UCSI_CAP_CABLE_DETAILS)
                                ucsi_partner_task(con, ucsi_check_cable, 1, HZ);
 
                        if (UCSI_CONSTAT_PWR_OPMODE(con->status.flags) ==
-                           UCSI_CONSTAT_PWR_OPMODE_PD)
+                           UCSI_CONSTAT_PWR_OPMODE_PD) {
                                ucsi_partner_task(con, ucsi_register_partner_pdos, 1, HZ);
+                               ucsi_partner_task(con, ucsi_check_connector_capability, 1, HZ);
+                       }
                } else {
                        ucsi_unregister_partner(con);
                }
                ucsi_register_device_pdos(con);
                ucsi_get_src_pdos(con);
                ucsi_check_altmodes(con);
+               ucsi_check_connector_capability(con);
        }
 
        trace_ucsi_register_port(con->num, &con->status);