diff --git a/src/main/java/gov/nasa/jpl/mbee/mdk/model/TableStructure.java b/src/main/java/gov/nasa/jpl/mbee/mdk/model/TableStructure.java
index 498196e7f..fa7abc8dd 100644
--- a/src/main/java/gov/nasa/jpl/mbee/mdk/model/TableStructure.java
+++ b/src/main/java/gov/nasa/jpl/mbee/mdk/model/TableStructure.java
@@ -112,9 +112,9 @@ else if (dgElement instanceof CallBehaviorAction) {
         }
         if (ts != null) {
             bnode = GeneratorUtils.findInitialNode(ts);
-            if (ts instanceof NamedElement) {
-                title = ((NamedElement) ts).getName();
-            }
+        }
+        if (dgElement instanceof NamedElement) {
+            title = ((NamedElement) dgElement).getName();
         }
     }