From d2f0eafa27e4aa0adaee2362ccc9bab8c658d955 Mon Sep 17 00:00:00 2001
From: guywatson <guy.watson@internetrix.com.au>
Date: Tue, 28 Oct 2014 09:24:40 +1100
Subject: [PATCH] Add an extension hook to the getIsModifiedOnStage method

---
 code/model/SiteTree.php | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/code/model/SiteTree.php b/code/model/SiteTree.php
index 696ded9c..553ce046 100644
--- a/code/model/SiteTree.php
+++ b/code/model/SiteTree.php
@@ -2808,8 +2808,11 @@ class SiteTree extends DataObject implements PermissionProvider,i18nEntityProvid
 		
 		$stageVersion = Versioned::get_versionnumber_by_stage('SiteTree', 'Stage', $this->ID);
 		$liveVersion =	Versioned::get_versionnumber_by_stage('SiteTree', 'Live', $this->ID);
-
-		return ($stageVersion && $stageVersion != $liveVersion);
+		
+		$isModified = ($stageVersion && $stageVersion != $liveVersion);
+		$this->extend('getIsModifiedOnStage', $isModified);
+		
+		return $isModified;
 	}
 	
 	/**