diff --git a/java/structures/WaveletTree.java b/java/structures/WaveletTree.java
index 3b58f3cf..b3795372 100644
--- a/java/structures/WaveletTree.java
+++ b/java/structures/WaveletTree.java
@@ -84,7 +84,7 @@ static int stablePartition(int[] a, int from, int to, Predicate<Integer> p) {
         }
         System.arraycopy(b1, 0, a, from, cnt1);
         System.arraycopy(b2, 0, a, from + cnt1, cnt2);
-        return cnt1;
+        return from + cnt1;
     }
 
     // Usage example