diff --git a/build/uk_master_test.xsl b/build/uk_master_test.xsl
index 3a49cd59..a226ef3e 100644
--- a/build/uk_master_test.xsl
+++ b/build/uk_master_test.xsl
@@ -30,9 +30,27 @@
 	<xsl:variable name="now" select="date:date-time()"/>
 	
 	<!--
-		Add a comment to the start of the output file.
+		Document root.
 	-->
 	<xsl:template match="/">
+		<xsl:call-template name="document.comment"/>
+		<xsl:apply-templates/>
+	</xsl:template>
+	
+	<!--
+		Document element.
+	-->
+	<xsl:template match="/md:EntitiesDescriptor">
+		<xsl:copy>
+			<xsl:call-template name="document.comment"/>
+			<xsl:apply-templates/>
+		</xsl:copy>
+	</xsl:template>
+
+	<!--
+		Comment to be added to the top of the document, and just inside the document element.
+	-->
+	<xsl:template name="document.comment">
 		<xsl:comment>
 			<xsl:text>&#10;&#9;U K   F E D E R A T I O N   M E T A D A T A&#10;</xsl:text>
 			<xsl:text>&#10;</xsl:text>
@@ -42,7 +60,6 @@
 			<xsl:value-of select="$now"/>
 			<xsl:text>&#10;</xsl:text>
 		</xsl:comment>
-		<xsl:apply-templates/>
 	</xsl:template>
 	
 	<!--