I'm trying to create a wafer map where each die would be colored differently based on a predetermined color code (i.e. FAIL TYPE 1 = color 1, FAIL TYPE 2 = color 2, PASS = color 3, etc.). Best I am able to get is something like this graph builder heat map:

In this example the color coding needs to be determined by column "CATEGORY" which is Character/Nominal and has values as PASS, FAIL1, FAIL2, FAIL3. However the way the graph builder appears to handle it is ignoring the actual values and making them kind of numeric/continuous, as can be seen above by "Count", 0-3, and the color scheme is automatically set to Blue-to-Red.
Code is shown below. Is there any way to specify the color scheme as needed? I am using JMP Pro 14.3. TIA!!
Graph Builder(
Size( 461, 400 ),
Variables( X( :X ), Y( :Y ), Color( :CATEGORY ) ),
Elements( Heatmap( X, Y, Legend( 1 ) ) ),
SendToReport(
Dispatch( {"Heatmap"}, "", OutlineBox, {Close( 0 )} ),
Dispatch( {}, "400", LegendBox, {Set Title( "Count" )} )
)
);