From a064903e908e774df704610a468d4c1efcbf3d44 Mon Sep 17 00:00:00 2001
From: Florian Verdonck <verdonckflorian@gmail.com>
Date: Sat, 8 Feb 2014 00:28:52 +0100
Subject: [PATCH] Updated request after feedback

---
 src/Umbraco.Core/Models/ITemplate.cs | 6 ++++++
 src/Umbraco.Core/Models/Template.cs  | 8 +++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/Umbraco.Core/Models/ITemplate.cs b/src/Umbraco.Core/Models/ITemplate.cs
index 2014ac9837..7fb4eac4cf 100644
--- a/src/Umbraco.Core/Models/ITemplate.cs
+++ b/src/Umbraco.Core/Models/ITemplate.cs
@@ -10,5 +10,11 @@
         /// </summary>
         /// <returns><see cref="RenderingEngine"/></returns>
         RenderingEngine GetTypeOfRenderingEngine();
+
+        /// <summary>
+        /// Set the mastertemplate
+        /// </summary>
+        /// <param name="masterTemplate"></param>
+        void SetMasterTemplate(ITemplate masterTemplate);
     }
 }
\ No newline at end of file
diff --git a/src/Umbraco.Core/Models/Template.cs b/src/Umbraco.Core/Models/Template.cs
index dc56700386..0b38c19a51 100644
--- a/src/Umbraco.Core/Models/Template.cs
+++ b/src/Umbraco.Core/Models/Template.cs
@@ -96,7 +96,7 @@ namespace Umbraco.Core.Models
         }
         
         [DataMember]
-        public Lazy<int> MasterTemplateId { get; set; }
+        internal Lazy<int> MasterTemplateId { get; set; }
 
         [DataMember]
         internal string MasterTemplateAlias
@@ -183,5 +183,11 @@ namespace Umbraco.Core.Models
             if (Key == Guid.Empty)
                 Key = Guid.NewGuid();
         }
+
+
+        public void SetMasterTemplate(ITemplate masterTemplate)
+        {
+            MasterTemplateId = new Lazy<int>(() => {return masterTemplate.Id});
+        }
     }
 }