Hydra 2.0 PrimsDirtied question

Hi,

I have a question regarding the new Hydra 2.0 framework.
I have a mesh with a material and an editor that can change some materialX value (for example input:base_color). I do it with

               const pxr::UsdAttribute &attribute = m_stage->GetAttributeAtPath(m_path);
                attribute.Set(value, m_timeCode);

I do receive the _PrimDirtied for the material, with the locator and the new datasourcehandle. But I was expecting the locator to be input:base_color rather than “material”.
As I got the “material” locator, it seems that I’m forced to reparse all the schema and recreate a fresh new material instead of just updating the new value.
Am I doing something wrong ?

Thank you,

Baptiste