diff --git a/compiler/src/process_ui_syntax.ts b/compiler/src/process_ui_syntax.ts
index 0cbd33e..5edf1a2 100644
--- a/compiler/src/process_ui_syntax.ts
+++ b/compiler/src/process_ui_syntax.ts
@@ -484,9 +484,10 @@ function createEntryNode(node: ts.SourceFile, context: ts.TransformationContext)
       return node;
     }
   } else {
-      const entryNode: ts.ExpressionStatement =
-        createEntryFunction(componentCollection.entryComponent, context);
-      return context.factory.updateSourceFile(node, [...node.statements, entryNode]);
+    const entryNode: ts.ExpressionStatement =
+      createEntryFunction(componentCollection.entryComponent ||
+        Array.from(componentCollection.previewComponent)[0], context);
+    return context.factory.updateSourceFile(node, [...node.statements, entryNode]);
   }
 }