diff --git a/compress/cstream.c b/compress/cstream.c
index 04d969c..283f145 100644
--- a/compress/cstream.c
+++ b/compress/cstream.c
@@ -6,7 +6,7 @@
 #include <string.h>
 
 enum b_status b_cstream_open(
-	struct b_stream *endpoint, const struct b_compression_function *func,
+	b_stream *endpoint, const struct b_compression_function *func,
 	enum b_compression_mode mode, struct b_cstream **out)
 {
 	size_t inbuf_size = 0, outbuf_size = 0;
diff --git a/compress/cstream.h b/compress/cstream.h
index 70c906d..7add9a6 100644
--- a/compress/cstream.h
+++ b/compress/cstream.h
@@ -3,6 +3,7 @@
 
 #include <blue/compress/function.h>
 #include <blue/core/ringbuffer.h>
+#include <blue/core/stream.h>
 #include <stddef.h>
 
 struct b_stream;
@@ -14,7 +15,7 @@ enum cstream_flags {
 
 struct b_cstream {
 	enum cstream_flags s_flags;
-	struct b_stream *s_endpoint;
+	b_stream *s_endpoint;
 	struct b_compressor *s_compressor;
 	/* s_in is the input buffer, and s_out is the output buffer.
 	 *