}
 }
 
-bool dml2_validate(const struct dc *in_dc, struct dc_state *context, bool fast_validate)
+bool dml2_validate(const struct dc *in_dc, struct dc_state *context, struct dml2_context *dml2, bool fast_validate)
 {
        bool out = false;
 
-       if (!(context->bw_ctx.dml2))
+       if (!dml2)
                return false;
-       dml2_apply_debug_options(in_dc, context->bw_ctx.dml2);
+       dml2_apply_debug_options(in_dc, dml2);
 
 
        /* Use dml_validate_only for fast_validate path */
 
  */
 bool dml2_validate(const struct dc *in_dc,
                                   struct dc_state *context,
+                                  struct dml2_context *dml2,
                                   bool fast_validate);
 
 /*
 
        bool out = false;
 
        if (dc->debug.using_dml2)
-               out = dml2_validate(dc, context, fast_validate);
+               out = dml2_validate(dc, context, context->bw_ctx.dml2, fast_validate);
        else
                out = dml1_validate(dc, context, fast_validate);
        return out;
 
 {
        bool out = false;
 
-       out = dml2_validate(dc, context, fast_validate);
+       out = dml2_validate(dc, context, context->bw_ctx.dml2, fast_validate);
 
        if (fast_validate)
                return out;
 
 {
        bool out = false;
 
-       out = dml2_validate(dc, context, fast_validate);
+       out = dml2_validate(dc, context, context->bw_ctx.dml2, fast_validate);
 
        if (fast_validate)
                return out;