BA comparison fails when comparing augmentation

We are suffering some issues comparing BA.

The corresponding model is

grouping flowspec-l3vpn-ipv6-routes {
container flowspec-l3vpn-ipv6-routes {
uses flowspec-ipv6-route {
augment "flowspec-route"

}
}
}

When FlowspecRouteBuilder#Equals

if (!e.getValue().equals(other.getAugmentation(e.getKey()))) {

First object contains Route distinguisher as augmentation, meanwhile the other contains it as children, therefore it fails.

Firs object creation

ROUTE = new FlowspecRouteBuilder().setKey(ROUTE_KEY)
.setPathId(PATH_ID).setFlowspec(FLOW_LIST)
.setAttributes(new AttributesBuilder().build())
.addAugmentation(FlowspecRoute1.class, new FlowspecRoute1Builder().setRouteDistinguisher(RD).build()).build();

Second one is readed from DS and converted using BindingToNormalizedNodeCodec->fromNormalizedNode