diff --git a/PostHogTests/PostHogMaskFlattenedContentTest.swift b/PostHogTests/PostHogMaskFlattenedContentTest.swift
index d5339c0ae..56aed5e73 100644
--- a/PostHogTests/PostHogMaskFlattenedContentTest.swift
+++ b/PostHogTests/PostHogMaskFlattenedContentTest.swift
@@ -7,7 +7,7 @@
 //  per-Text masking heuristics can't see the text inside. These tests host real SwiftUI and
 //  check the collected mask rects geometrically.
 
-#if os(iOS) && canImport(SwiftUI)
+#if os(iOS) && canImport(SwiftUI) && (!SWIFT_PACKAGE || SessionReplay)
     import Foundation
     @testable import PostHog
     import SwiftUI