static int
 mlxsw_sp_tclass_congestion_enable(struct mlxsw_sp_port *mlxsw_sp_port,
                                  int tclass_num, u32 min, u32 max,
-                                 u32 probability, bool is_ecn)
+                                 u32 probability, bool is_wred, bool is_ecn)
 {
        char cwtpm_cmd[MLXSW_REG_CWTPM_LEN];
        char cwtp_cmd[MLXSW_REG_CWTP_LEN];
                return err;
 
        mlxsw_reg_cwtpm_pack(cwtpm_cmd, mlxsw_sp_port->local_port, tclass_num,
-                            MLXSW_REG_CWTP_DEFAULT_PROFILE, true, is_ecn);
+                            MLXSW_REG_CWTP_DEFAULT_PROFILE, is_wred, is_ecn);
 
        return mlxsw_reg_write(mlxsw_sp->core, MLXSW_REG(cwtpm), cwtpm_cmd);
 }
        prob = DIV_ROUND_UP(prob, 1 << 16);
        min = mlxsw_sp_bytes_cells(mlxsw_sp, p->min);
        max = mlxsw_sp_bytes_cells(mlxsw_sp, p->max);
-       return mlxsw_sp_tclass_congestion_enable(mlxsw_sp_port, tclass_num, min,
-                                                max, prob, p->is_ecn);
+       return mlxsw_sp_tclass_congestion_enable(mlxsw_sp_port, tclass_num,
+                                                min, max, prob,
+                                                !p->is_nodrop, p->is_ecn);
 }
 
 static void