1 files changed, 2 insertions, 2 deletions
diff --git a/model-evaluation/src/main/java/ai/vespa/models/evaluation/LazyArrayContext.java b/model-evaluation/src/main/java/ai/vespa/models/evaluation/LazyArrayContext.java
index 40a0a1be5fc..d97235d11d2 100644
--- a/model-evaluation/src/main/java/ai/vespa/models/evaluation/LazyArrayContext.java
+++ b/model-evaluation/src/main/java/ai/vespa/models/evaluation/LazyArrayContext.java
@@ -293,8 +293,8 @@ public final class LazyArrayContext extends Context implements ContextIndex {
                 ReferenceNode reference = (ReferenceNode) node;
                 if (reference.getArguments().size() > 0) {
                     if (reference.getArguments().expressions().get(0) instanceof ConstantNode) {
-                        ConstantNode constantNode = (ConstantNode) reference.getArguments().expressions().get(0);
-                        return Optional.of(stripQuotes(constantNode.sourceString()));
+                        ExpressionNode constantNode = reference.getArguments().expressions().get(0);
+                        return Optional.of(stripQuotes(constantNode.toString()));
                     }
                     if (reference.getArguments().expressions().get(0) instanceof ReferenceNode) {
                         ReferenceNode referenceNode = (ReferenceNode) reference.getArguments().expressions().get(0);