Resource Management Workspace - issue with grouping by new planning attribute

[DanielCordick]

I have added location as a planning attribute however my group by functionality is not working in RMW, not sure what im missing?? any one every got this to work??

The group by button, is attached to the RMWorkspaceServiceSNC script include which i have extended to include Primary location. 

ive also modified the UX Client Script include RMWorkspaceGroupByUtils and included my planning attribute.

const getGroupData = (asmtRow, userRow, groupByItem, emptyGroupName) => {
const primaryAttributes = ['group_resource', 'role', 'skill', 'u_location']; <<<<------------ ADDED u_location to map 
let groudId, groupLabel;
if(primaryAttributes.indexOf(groupByItem.id) >= 0) {
//We are fetching primaryAttributes values from user row.
groudId = userRow.task[groupByItem.id].value? userRow.task[groupByItem.id].value : "emptyId";
groupLabel = userRow.task[groupByItem.id].displayValue? userRow.task[groupByItem.id].displayValue : emptyGroupName;
}
else {
//We are fetching 'owner' and 'parent item' values from assignment row.
groudId = (asmtRow[groupByItem.id] && asmtRow[groupByItem.id].value) ? asmtRow[groupByItem.id].value : "emptyId";
groupLabel = (asmtRow[groupByItem.id] && asmtRow[groupByItem.id].displayValue) ? asmtRow[groupByItem.id].displayValue : emptyGroupName;
}
return {groudId, groupLabel};
};