diff --git a/Tools/AGSHtml.m b/Tools/AGSHtml.m
index 2b192808f..55ccddeec 100644
--- a/Tools/AGSHtml.m
+++ b/Tools/AGSHtml.m
@@ -2740,7 +2740,7 @@ - (void) outputUnit: (GSXMLNode*)node to: (NSMutableString*)buf
       [dot appendFormat: @"digraph class_%@ {\n", cNam];
       [dot appendString: @" rankdir = \"TB\";\n"];
       [dot appendString: @" {\n"];
-      [dot appendString: @"   node [margin=0 fontcolor=blue"
+      [dot appendString: @"   node [class=figure margin=0 fontcolor=blue"
 	@" fontsize=24 width=0.5 shape=rectangle style=filled]\n"];
       if (sNam)
 	{