handle tip nodes when counting connected internal nodes

beast-dev · Oct 31, 2023 · 37064a5 · 37064a5
1 parent 61de19f
commit 37064a5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/dr/evomodel/treedatalikelihood/discrete/NodeHeightBounds.java b/src/dr/evomodel/treedatalikelihood/discrete/NodeHeightBounds.java
@@ -67,10 +67,12 @@ public int[] getConnectedParameterIndices(int index) {
             nodeCount++;
         }
         int[] connectedIndices = new int[nodeCount];
+        int nodeIndex = 0;
         for (int i = 0; i < treeModel.getChildCount(nodeRef); i++) {
             NodeRef childNode = treeModel.getChild(nodeRef, i);
             if (!treeModel.isExternal(childNode)) {
-                connectedIndices[i] = childNode.getNumber() - externalNodeCount;
+                connectedIndices[nodeIndex] = childNode.getNumber() - externalNodeCount;
+                nodeIndex++;
             }
         }
         if (!treeModel.isRoot(nodeRef)) {