diff --git a/trunk/src/com/jpexs/asdec/abc/types/traits/Traits.java b/trunk/src/com/jpexs/asdec/abc/types/traits/Traits.java
index 0d12147cb..091b66700 100644
--- a/trunk/src/com/jpexs/asdec/abc/types/traits/Traits.java
+++ b/trunk/src/com/jpexs/asdec/abc/types/traits/Traits.java
@@ -68,7 +68,7 @@ public class Traits {
                }
             }
             if (traits[t] instanceof TraitClass) {
-               plus = Highlighting.hilighClass(plus, h);
+               plus = Highlighting.hilighClass(plus, ((TraitClass)traits[t]).class_info);
             } else {
                plus = Highlighting.hilighTrait(plus, h);
             }