intel_dp_mst_configure(intel_dp);
 
-       /*
-        * TODO: Reset link params when switching to MST mode, until MST
-        * supports link training fallback params.
-        */
-       if (intel_dp->reset_link_params || intel_dp->is_mst) {
+       if (intel_dp->reset_link_params) {
                intel_dp_reset_link_params(intel_dp);
                intel_dp->reset_link_params = false;
        }
 
 
        new_rate = intel_dp_common_rate(intel_dp, rate_index - 1);
 
+       /* TODO: Make switching from UHBR to non-UHBR rates work. */
+       if (drm_dp_is_uhbr_rate(current_rate) != drm_dp_is_uhbr_rate(new_rate))
+               return -1;
+
        return new_rate;
 }
 
        int new_link_rate;
        int new_lane_count;
 
-       /*
-        * TODO: Enable fallback on MST links once MST link compute can handle
-        * the fallback params.
-        */
-       if (intel_dp->is_mst) {
-               lt_err(intel_dp, DP_PHY_DPRX, "Link Training Unsuccessful\n");
-               return -1;
-       }
-
        if (intel_dp_is_edp(intel_dp) && !intel_dp->use_max_params) {
                lt_dbg(intel_dp, DP_PHY_DPRX,
                       "Retrying Link training for eDP with max parameters\n");