From f4f83325a778041aca03f7fe5a912434db78c81d Mon Sep 17 00:00:00 2001
From: Andras Slemmer <andras.slemmer@r3cev.com>
Date: Tue, 7 Mar 2017 16:38:26 +0000
Subject: [PATCH] client: Address PR 290 comments

---
 client/src/main/kotlin/net/corda/client/mock/Generator.kt | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/client/src/main/kotlin/net/corda/client/mock/Generator.kt b/client/src/main/kotlin/net/corda/client/mock/Generator.kt
index c0fba098e1..660fd8054d 100644
--- a/client/src/main/kotlin/net/corda/client/mock/Generator.kt
+++ b/client/src/main/kotlin/net/corda/client/mock/Generator.kt
@@ -178,9 +178,10 @@ fun <A : Any> Generator.Companion.pickN(number: Int, list: List<A>) = Generator<
     val mask = BitSet(list.size)
     val size = Math.min(list.size, number)
     for (i in 0..size - 1) {
+        // mask[i] = 1 desugars into mask.set(i, 1), which sets a range instead of a bit
         mask[i] = true
     }
-    for (i in 0..size - 1) {
+    for (i in 0..list.size - 1) {
         val bit = mask[i]
         val swapIndex = i + it.nextInt(size - i)
         mask[i] = mask[swapIndex]