Clean up a bit of the ShrinkTree stuff, particularly for shrinking

1 files changed, 1 insertions, 2 deletions

diff --git a/src/main/java/au/id/zancanaro/javacheck/Generators.java b/src/main/java/au/id/zancanaro/javacheck/Generators.java
index e5699ac..9730bac 100644
--- a/src/main/java/au/id/zancanaro/javacheck/Generators.java
+++ b/src/main/java/au/id/zancanaro/javacheck/Generators.java
@@ -29,7 +29,7 @@ public final class Generators {
      * removed
      */
     public static <T> Generator<T> noShrink(Generator<T> gen) {
-        return (random, size) -> ShrinkTree.pure(gen.generate(random, size).getValue());
+        return gen.withShrinkStrategy(value -> Iterators.emptyIterator());
     }
 
     @SafeVarargs
@@ -230,7 +230,6 @@ public final class Generators {
                 chars[i++] = c;
             }
             return String.valueOf(chars);
-//            return new String(chars);
         });
     }
 }