diff --git a/Sources/KlaviyoUI/KlaviyoWebView/KlaviyoWebWrapperViewController.swift b/Sources/KlaviyoUI/KlaviyoWebView/KlaviyoWebWrapperViewController.swift
index a4dec177..4f9183b2 100644
--- a/Sources/KlaviyoUI/KlaviyoWebView/KlaviyoWebWrapperViewController.swift
+++ b/Sources/KlaviyoUI/KlaviyoWebView/KlaviyoWebWrapperViewController.swift
@@ -135,7 +135,7 @@ func createKlaviyoWebPreview(url: URL, style: KlaviyoWebWrapperStyle) -> UIViewC
 @available(iOS 17.0, *)
 #Preview("Default style") {
     let url = URL(string: "https://www.google.com")!
-    createKlaviyoWebPreview(url: url, style: .default)
+    return createKlaviyoWebPreview(url: url, style: .default)
 }
 
 @available(iOS 17.0, *)
@@ -147,6 +147,6 @@ func createKlaviyoWebPreview(url: URL, style: KlaviyoWebWrapperStyle) -> UIViewC
         cornerRadius: 24,
         shadowStyle: .default)
 
-    createKlaviyoWebPreview(url: url, style: style)
+    return createKlaviyoWebPreview(url: url, style: style)
 }
 #endif