diff --git a/app/Engine/Engine.php b/app/Engine/Engine.php
index e2493f7..0f5c9d1 100644
--- a/app/Engine/Engine.php
+++ b/app/Engine/Engine.php
@@ -91,10 +91,8 @@ class Engine
 		$documents = array_values($documents);
 
 		// Pass it all in and compile
-		$document = $this->compileDocuments($documents, $options);
-
-		// Set metadata in output
-		$document->setChildByPath('Metadata.Stack', $files);
+		$this->compileDocuments($documents, $options);
+		$document = $this->getOutputDocument();
 
 		// Remove cfnpp block, that doesn't go to CloudFormation
 		$cfnpp = $document->getChildByName('cfnpp');