diff --git a/Tests/HTMLKitTests/CSSSelectorParserTests.m b/Tests/HTMLKitTests/CSSSelectorParserTests.m
index 1966e44..545407e 100644
--- a/Tests/HTMLKitTests/CSSSelectorParserTests.m
+++ b/Tests/HTMLKitTests/CSSSelectorParserTests.m
@@ -7,7 +7,7 @@
 //
 
 #import <XCTest/XCTest.h>
-#import <HTMLKit/HTMLKit.h>
+#import "HTMLKit.h"
 #import "CSSSelectorTest.h"
 #import "CSSSelectorParser.h"
 
diff --git a/Tests/HTMLKitTests/HTML5LibTokenizerTest.m b/Tests/HTMLKitTests/HTML5LibTokenizerTest.m
index b46f8e4..b4f63b2 100644
--- a/Tests/HTMLKitTests/HTML5LibTokenizerTest.m
+++ b/Tests/HTMLKitTests/HTML5LibTokenizerTest.m
@@ -7,6 +7,7 @@
 //
 
 #import "HTML5LibTokenizerTest.h"
+#import "HTMLOrderedDictionary.h"
 #import "HTMLTokenizerStates.h"
 #import "HTMLTokens.h"
 #import "HTMLKitTestUtil.h"