1 files changed, 2 insertions, 2 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/OrderedTensorType.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/OrderedTensorType.java
index ceb9a27924d..eee60d56c55 100644
--- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/OrderedTensorType.java
+++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/OrderedTensorType.java
@@ -96,14 +96,14 @@ public class OrderedTensorType {
      * so that they are correctly laid out in memory for Vespa.
      * Used when importing tensors.
      */
-    public int toDirectIndex(int index) {
+    public long toDirectIndex(int index) {
         if (dimensions.size() == 0) {
             return 0;
         }
         if (dimensionMap == null)  {
             throw new IllegalArgumentException("Dimension map is not available");
         }
-        int directIndex = 0;
+        long directIndex = 0;
         long rest = index;
         for (int i = 0; i < dimensions.size(); ++i) {
             long address = rest / innerSizesOriginal[i];