1 files changed, 11 insertions, 4 deletions
diff --git a/Tools/TestWebKitAPI/Tests/WebKit2ObjC/CustomProtocolsInvalidScheme_Bundle.cpp b/Tools/TestWebKitAPI/Tests/WebKit2ObjC/CustomProtocolsInvalidScheme_Bundle.cpp
index ee72f87e7..287f32ff5 100644
--- a/Tools/TestWebKitAPI/Tests/WebKit2ObjC/CustomProtocolsInvalidScheme_Bundle.cpp
+++ b/Tools/TestWebKitAPI/Tests/WebKit2ObjC/CustomProtocolsInvalidScheme_Bundle.cpp
@@ -25,8 +25,10 @@
 
 #include "config.h"
 
+#if WK_HAVE_C_SPI
+
 #include "InjectedBundleTest.h"
-#include <WebKit2/WKBundlePage.h>
+#include <WebKit/WKBundlePage.h>
 
 namespace TestWebKitAPI {
 
@@ -45,15 +47,20 @@ public:
     }
 
 private:
-    virtual void didCreatePage(WKBundleRef, WKBundlePageRef bundlePage) OVERRIDE
+    virtual void didCreatePage(WKBundleRef, WKBundlePageRef bundlePage) override
     {
-        WKBundlePagePolicyClient policyClient;
+        WKBundlePagePolicyClientV0 policyClient;
         memset(&policyClient, 0, sizeof(policyClient));
+
+        policyClient.base.version = 0;
         policyClient.decidePolicyForNavigationAction = decidePolicyForNavigationAction;
-        WKBundlePageSetPolicyClient(bundlePage, &policyClient);
+        
+        WKBundlePageSetPolicyClient(bundlePage, &policyClient.base);
     }
 };
 
 static InjectedBundleTest::Register<CustomProtocolInvalidSchemeTest> registrar("CustomProtocolInvalidSchemeTest");
 
 } // namespace TestWebKitAPI
+
+#endif