diff --git a/exchange/bitswap/testutils.go b/exchange/bitswap/testutils.go
index b8763952c..48cb11a45 100644
--- a/exchange/bitswap/testutils.go
+++ b/exchange/bitswap/testutils.go
@@ -23,7 +23,7 @@ func NewSessionGenerator(
 		net:    net,
 		rs:     rs,
 		seq:    0,
-		ctx:    ctx,
+		ctx:    ctx, // TODO take ctx as param to Next, Instances
 		cancel: cancel,
 	}
 }