Attribute dimensions in FlexibleDataGridDefinitionBuilder?

Summary:

Do FlexibleDataGridDefinitionBuilders allow the use of attribute dimensions?

Content (please ensure you mask any confidential information):

I've been told multiple times that I should stop using a DataGridDefinitionBuilder and start using FlexibleDataGridDefinitionBuilder wherever I can as it is better. I was just writing a new rule and figured I would try to start testing the use of the Flexible version but I need to use an attribute dimension in this definition and I'm getting an error that the dimension doesn't exist in the application. When I converted it back to the standard version, it works. Is there something special I need to do to get the flexible version to work with attribute