diff --git a/src/loader/head-transformer.ts b/src/loader/head-transformer.ts
index ab81f65..acc2c9c 100644
--- a/src/loader/head-transformer.ts
+++ b/src/loader/head-transformer.ts
@@ -19,15 +19,18 @@ function transformerMarko5(path, t): void {
     t.identifier("out"),
     t.identifier("___renderAssets")
   );
-  path.replaceWith(
-    t.markoScriptlet([
-      t.expressionStatement(
-        t.logicalExpression(
-          "&&",
-          renderAssetsMember,
-          t.callExpression(renderAssetsMember, [])
+  path
+    .get("body")
+    .pushContainer(
+      "body",
+      t.markoScriptlet([
+        t.expressionStatement(
+          t.logicalExpression(
+            "&&",
+            renderAssetsMember,
+            t.callExpression(renderAssetsMember, [])
+          )
         )
-      )
-    ])
-  );
+      ])
+    );
 }