From 5b79292fc5128ca17202b5e36fa16d3154c06f26 Mon Sep 17 00:00:00 2001
From: Anna Koskinen <Ansku@users.noreply.github.com>
Date: Fri, 5 Mar 2021 13:11:30 +0200
Subject: [PATCH] Fixed input to ensureAvailabilty (#12208) (#12224)

Authored-by: Oliver Buchien <30312325+BuchienDP@users.noreply.github.com>
---
 .../java/com/vaadin/client/ui/treegrid/TreeGridConnector.java | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/client/src/main/java/com/vaadin/client/ui/treegrid/TreeGridConnector.java b/client/src/main/java/com/vaadin/client/ui/treegrid/TreeGridConnector.java
index 253e03a80c8..54f7a112d48 100644
--- a/client/src/main/java/com/vaadin/client/ui/treegrid/TreeGridConnector.java
+++ b/client/src/main/java/com/vaadin/client/ui/treegrid/TreeGridConnector.java
@@ -208,7 +208,7 @@ public void dataRemoved(int firstRowIndex, int numberOfRows) {
                             .getVisibleRowRange();
                     getDataSource().ensureAvailability(
                             visibleRowRange.getStart(),
-                            visibleRowRange.getEnd());
+                            visibleRowRange.length());
                 }
                 checkExpand();
             }
@@ -222,7 +222,7 @@ public void dataAdded(int firstRowIndex, int numberOfRows) {
                             .getVisibleRowRange();
                     getDataSource().ensureAvailability(
                             visibleRowRange.getStart(),
-                            visibleRowRange.getEnd());
+                            visibleRowRange.length());
                 }
                 checkExpand();
             }