@@ -170,21 +170,21 @@
 		lengths[i] = 5;
 
 	fixedDistTree = of_huffman_tree_construct(lengths, 32);
 }
 
-+ (instancetype)streamWithStream: (OFStream *)stream
++ (instancetype)streamWithStream: (OF_KINDOF(OFStream *))stream
 {
 	return [[[self alloc] initWithStream: stream] autorelease];
 }
 
 - (instancetype)init
 {
 	OF_INVALID_INIT_METHOD
 }
 
-- (instancetype)initWithStream: (OFStream *)stream
+- (instancetype)initWithStream: (OF_KINDOF(OFStream *))stream
 {
 	self = [super init];
 
 	@try {
 		_stream = [stream retain];