1 files changed, 7 insertions, 1 deletions
diff --git a/pw_transfer/java/main/dev/pigweed/pw_transfer/ReadTransfer.java b/pw_transfer/java/main/dev/pigweed/pw_transfer/ReadTransfer.java
index 15e3cd321..aa9389b77 100644
--- a/pw_transfer/java/main/dev/pigweed/pw_transfer/ReadTransfer.java
+++ b/pw_transfer/java/main/dev/pigweed/pw_transfer/ReadTransfer.java
@@ -52,6 +52,7 @@ class ReadTransfer extends Transfer<byte[]> {
   private int lastReceivedOffset = 0;
 
   ReadTransfer(int resourceId,
+      int sessionId,
       ProtocolVersion desiredProtocolVersion,
       TransferInterface transferManager,
       TransferTimeoutSettings timeoutSettings,
@@ -59,6 +60,7 @@ class ReadTransfer extends Transfer<byte[]> {
       Consumer<TransferProgress> progressCallback,
       BooleanSupplier shouldAbortCallback) {
     super(resourceId,
+        sessionId,
         desiredProtocolVersion,
         transferManager,
         timeoutSettings,
@@ -68,6 +70,10 @@ class ReadTransfer extends Transfer<byte[]> {
     this.windowEndOffset = parameters.maxPendingBytes();
   }
 
+  final TransferParameters getParametersForTest() {
+    return parameters;
+  }
+
   @Override
   State getWaitingForDataState() {
     return new ReceivingData();
@@ -183,7 +189,7 @@ class ReadTransfer extends Transfer<byte[]> {
 
     ByteBuffer result = ByteBuffer.allocate(totalDataSize);
     dataChunks.forEach(result::put);
-    getFuture().set(result.array());
+    set(result.array());
   }
 
   private VersionedChunk prepareTransferParameters(boolean extend) {