diff --git a/safari-extension/safari-extension-content-tests.ts b/safari-extension/safari-extension-content-tests.ts
index b58735b9d..067e90f05 100644
--- a/safari-extension/safari-extension-content-tests.ts
+++ b/safari-extension/safari-extension-content-tests.ts
@@ -1,4 +1,4 @@
-/// <reference path="safari-content.d.ts" />
+/// <reference path="safari-extension-content.d.ts" />
 
 // https://developer.apple.com/library/safari/documentation/Tools/Conceptual/SafariExtensionGuide/AddingExtensionToolbars/AddingExtensionToolbars.html#//apple_ref/doc/uid/TP40009977-CH5-SW7
 var theBody = document.body;
diff --git a/safari-extension/safari-extension-tests.ts b/safari-extension/safari-extension-tests.ts
index b4e5cfd60..0336c4411 100644
--- a/safari-extension/safari-extension-tests.ts
+++ b/safari-extension/safari-extension-tests.ts
@@ -1,4 +1,4 @@
-/// <reference path="safari.d.ts" />
+/// <reference path="safari-extension.d.ts" />
 
 // https://developer.apple.com/library/safari/documentation/Tools/Conceptual/SafariExtensionGuide/AccessingResourcesWithinYourExtensionFolder/AccessingResourcesWithinYourExtensionFolder.html#//apple_ref/doc/uid/TP40009977-CH18-SW2
 var img = document.createElement("img");