diff --git a/HTMLKitTests/HTMLTokenizerTests.m b/HTMLKitTests/HTMLTokenizerTests.m
index ae79952..80a70b3 100644
--- a/HTMLKitTests/HTMLTokenizerTests.m
+++ b/HTMLKitTests/HTMLTokenizerTests.m
@@ -8,20 +8,36 @@
 
 #import <Cocoa/Cocoa.h>
 #import <XCTest/XCTest.h>
+#import "HTMLKitTests.h"
 #import "HTML5LibTest.h"
 
-@interface HTMLTokenizerTests : XCTestCase
+#import "HTMLTokenizer.h"
+#import "HTMLTokenizerStates.h"
+#import "HTMLTokens.h"
+
+@implementation HTMLParseErrorToken (Testing)
+
+- (BOOL)isEqual:(id)object
+{
+	return [object isKindOfClass:[HTMLParseErrorToken class]];
+}
+
+@end
+
+@interface HTMLTokenizerTests : HTMLKitTests
 
 @end
 
 @implementation HTMLTokenizerTests
 
-- (void)setUp {
+- (void)setUp
+{
     [super setUp];
     // Put setup code here. This method is called before the invocation of each test method in the class.
 }
 
-- (void)tearDown {
+- (void)tearDown
+{
     // Put teardown code here. This method is called after the invocation of each test method in the class.
     [super tearDown];
 }