Author: mdryakhlenkov Date: 2007-07-03 07:26:22 -0400 (Tue, 03 Jul 2007) New Revision: 2252 Modified: trunk/hibernatetools/plugins/org.jboss.tools.hibernate.ui.veditor/src/org/jboss/tools/hibernate/veditor/editors/parts/OrmShapeEditPart.java Log: Modified: trunk/hibernatetools/plugins/org.jboss.tools.hibernate.ui.veditor/src/org/jboss/tools/hibernate/veditor/editors/parts/OrmShapeEditPart.java =================================================================== --- trunk/hibernatetools/plugins/org.jboss.tools.hibernate.ui.veditor/src/org/jboss/tools/hibernate/veditor/editors/parts/OrmShapeEditPart.java 2007-07-03 11:25:42 UTC (rev 2251) +++ trunk/hibernatetools/plugins/org.jboss.tools.hibernate.ui.veditor/src/org/jboss/tools/hibernate/veditor/editors/parts/OrmShapeEditPart.java 2007-07-03 11:26:22 UTC (rev 2252) @@ -31,6 +31,7 @@ import org.hibernate.mapping.PersistentClass; import org.hibernate.mapping.Property; import org.hibernate.mapping.RootClass; +import org.hibernate.mapping.SingleTableSubclass; import org.hibernate.mapping.Table; import org.jboss.tools.hibernate.veditor.editors.figures.RoundLineBorder; import org.jboss.tools.hibernate.veditor.editors.figures.TitleFigure; @@ -53,7 +54,7 @@ TitleLabel label = new TitleLabel(); String text = ""; Object element = getCastedModel().getOrmElement(); - if (element instanceof PersistentClass) { + if (element instanceof RootClass) { text = ormLabelProvider.getText((PersistentClass)element); } else if (element instanceof Table) { Table table = (Table)element;