diff --git a/HTMLKit/HTMLTokenizer.h b/HTMLKit/HTMLTokenizer.h
index 30726a0..faeb866 100644
--- a/HTMLKit/HTMLTokenizer.h
+++ b/HTMLKit/HTMLTokenizer.h
@@ -15,4 +15,6 @@
*/
@interface HTMLTokenizer : NSObject
+- (instancetype)initWithString:(NSString *)string;
+
@end
diff --git a/HTMLKit/HTMLTokenizer.m b/HTMLKit/HTMLTokenizer.m
index b6ce87b..66eb95d 100644
--- a/HTMLKit/HTMLTokenizer.m
+++ b/HTMLKit/HTMLTokenizer.m
@@ -53,7 +53,7 @@
#pragma mark - Lifecycle
-- (instancetype)init
+- (instancetype)initWithString:(NSString *)string
{
self = [super init];
if (self) {
@@ -62,6 +62,8 @@
[self setupStateMachine];
_tokens = [NSMutableArray new];
+
+ _inputStreamReader = [[HTMLInputStreamReader alloc] initWithString:string];
}
return self;
}