diff --git a/src/dorkbox/network/pipeline/AeronInput.java b/src/dorkbox/network/pipeline/AeronInput.java
index 8d47b71f..4f734aa1 100644
--- a/src/dorkbox/network/pipeline/AeronInput.java
+++ b/src/dorkbox/network/pipeline/AeronInput.java
@@ -45,10 +45,8 @@ import com.esotericsoftware.kryo.KryoException;
 import com.esotericsoftware.kryo.io.Input;
 import com.esotericsoftware.kryo.util.Util;
 
-import io.netty.buffer.ByteBuf;
-
 /**
- * An {@link InputStream} which reads data from a {@link ByteBuf}.
+ * An {@link InputStream} which reads data from a {@link DirectBuffer}.
  * <p/>
  * A read operation against this stream will occur at the {@code readerIndex}
  * of its underlying buffer and the {@code readerIndex} will increase during
diff --git a/src/dorkbox/network/pipeline/AeronOutput.java b/src/dorkbox/network/pipeline/AeronOutput.java
index 91af4daf..1c2fdf62 100644
--- a/src/dorkbox/network/pipeline/AeronOutput.java
+++ b/src/dorkbox/network/pipeline/AeronOutput.java
@@ -44,10 +44,8 @@ import org.agrona.concurrent.UnsafeBuffer;
 import com.esotericsoftware.kryo.KryoException;
 import com.esotericsoftware.kryo.io.Output;
 
-import io.netty.buffer.ByteBuf;
-
 /**
- * An {@link OutputStream} which writes data to a {@link ByteBuf}.
+ * An {@link OutputStream} which writes data to a {@link MutableDirectBuffer}.
  * <p>
  * A write operation against this stream will occur at the {@code writerIndex}
  * of its underlying buffer and the {@code writerIndex} will increase during