fix: document should never be null (#92)

sanity-io · Mar 18, 2024 · 1c4cdd3 · 1c4cdd3
1 parent 03d787b
commit 1c4cdd3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/Iframe.tsx b/src/Iframe.tsx
@@ -70,6 +70,7 @@ const MotionFlex = motion(Flex)
 
 export interface IframeProps {
   document: {
+    displayed: SanityDocument
     draft: SanityDocument | null
     published: SanityDocument | null
   }
@@ -78,7 +79,7 @@ export interface IframeProps {
 
 export function Iframe(props: IframeProps) {
   const {document, options} = props
-  const draft = document.draft || document.published
+  const draft = document.draft || document.published || document.displayed
 
   const {defaultSize = DEFAULT_SIZE, reload, attributes, showDisplayUrl = true, key} = options