From aa8b1e6bdba7c3a4577aaee7ef2051e970efd2d3 Mon Sep 17 00:00:00 2001
From: Andy Magee <andyfmagee@gmail.com>
Date: Wed, 27 Sep 2023 12:20:56 -0700
Subject: [PATCH] Allow non-DNA

---
 .../discrete/AbstractLogAdditiveSubstitutionModelGradient.java  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/dr/evomodel/treedatalikelihood/discrete/AbstractLogAdditiveSubstitutionModelGradient.java b/src/dr/evomodel/treedatalikelihood/discrete/AbstractLogAdditiveSubstitutionModelGradient.java
index 9ccbf1195f..7a37548980 100644
--- a/src/dr/evomodel/treedatalikelihood/discrete/AbstractLogAdditiveSubstitutionModelGradient.java
+++ b/src/dr/evomodel/treedatalikelihood/discrete/AbstractLogAdditiveSubstitutionModelGradient.java
@@ -204,7 +204,7 @@ public double[] getGradientLogDensity() {
             double[] qQPlus = getQQPlus(eigenVectors, inverseEigenVectors, index);
             double[] qPlusQ = getQPlusQ(eigenVectors, inverseEigenVectors, index);
 
-            double[] generator = new double[16];
+            double[] generator = new double[stateCount * stateCount];
             substitutionModel.getInfinitesimalMatrix(generator);
 
             for (int m = 0; m < stateCount; ++m) {