1 files changed, 2 insertions, 1 deletions
diff --git a/kgramstats.cpp b/kgramstats.cpp
index af24606..48e4b06 100644
--- a/kgramstats.cpp
+++ b/kgramstats.cpp

@@ -428,7 +428,8 @@ std::string kgramstats::randomSentence(int n)
      {
        while ((cur.size() > 1) && (cuts > 0))
        {
-          if ((rand() % cuts) > (maxK - cur.size() + 1))
+          int under = maxK - cur.size() + 1;
+          if ((rand() % cuts) > (under * under))
          {
            cur.pop_front();
            cuts--;