diff --git a/src/dorkbox/network/connection/streaming/AeronWriter.kt b/src/dorkbox/network/connection/streaming/AeronWriter.kt
index 8486fc00..5c86a2e7 100644
--- a/src/dorkbox/network/connection/streaming/AeronWriter.kt
+++ b/src/dorkbox/network/connection/streaming/AeronWriter.kt
@@ -18,4 +18,4 @@ package dorkbox.network.connection.streaming
 
 import dorkbox.network.serialization.AeronOutput
 
-class AeronWriter: StreamingWriter, AeronOutput()
+class AeronWriter(size: Int): StreamingWriter, AeronOutput(size)
diff --git a/src/dorkbox/network/connection/streaming/StreamingManager.kt b/src/dorkbox/network/connection/streaming/StreamingManager.kt
index e761afe1..284fa83d 100644
--- a/src/dorkbox/network/connection/streaming/StreamingManager.kt
+++ b/src/dorkbox/network/connection/streaming/StreamingManager.kt
@@ -130,7 +130,8 @@ internal class StreamingManager<CONNECTION : Connection>(
                     streamingDataInMemory[streamId] = FileWriter(tempFileLocation)
                 } else {
                     endPoint.logger.info { "Saving streaming data [${streamId}] in memory" }
-                    streamingDataInMemory[streamId] = AeronWriter()
+                    // .toInt is safe because we know the total size is < than maxStreamSizeInMemoryInBytes
+                    streamingDataInMemory[streamId] = AeronWriter(message.totalSize.toInt())
                 }
 
                 // this must be last