From 88828bd004ccba13e227f732106ab30c3731837f Mon Sep 17 00:00:00 2001
From: Mikolaj Konarski <mikolaj.konarski@funktory.com>
Date: Mon, 1 Dec 2025 18:38:56 +0100
Subject: Simplify genShRwithTarget a little

---
 test/Gen.hs | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/test/Gen.hs b/test/Gen.hs
index b10e763..4f5fe96 100644
--- a/test/Gen.hs
+++ b/test/Gen.hs
@@ -78,9 +78,8 @@ genShRwithTarget targetMax sn = do
         dims <- genDims m (if dim == 0 then 0 else tgt `div` dim)
         return (dim :$: dims)
   dims <- genDims sn targetSize
-  let dimsL = toList dims
-      maxdim = maximum dimsL
-      cap = binarySearch (`div` 2) 1 maxdim (\cap' -> product (min cap' <$> dimsL) <= targetSize)
+  let maxdim = maximum dims
+      cap = binarySearch (`div` 2) 1 maxdim (\cap' -> shrSize (min cap' <$> dims) <= targetSize)
   shuffleShR (min cap <$> dims)
 
 -- | Example: given 3 and 7, might return:
-- 
cgit v1.2.3-70-g09d2