From 90ae32dc51bc37db40766eafcf5c0b434442d78c Mon Sep 17 00:00:00 2001
From: Jeremy Shipman <jeremy@burnbright.net>
Date: Mon, 6 Oct 2014 15:19:56 +1300
Subject: [PATCH] Check if dataRecord is a SiteTree to prevent errors

---
 code/BetterNavigator.php | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/code/BetterNavigator.php b/code/BetterNavigator.php
index 471fdcf..72e26a7 100755
--- a/code/BetterNavigator.php
+++ b/code/BetterNavigator.php
@@ -26,7 +26,7 @@ class BetterNavigator extends DataExtension {
 		$isDev = Director::isDev();
 
 		if($isDev || Permission::check('CMS_ACCESS_CMSMain') || Permission::check('VIEW_DRAFT_CONTENT')) {			
-			if($this->owner && $this->owner->dataRecord) {
+			if($this->owner && $this->owner->dataRecord && $this->owner->dataRecord instanceof SiteTree) {
 
 				//Get SilverStripeNavigator links & stage info (CMS/Stage/Live/Archive)
 				$nav = array();
@@ -61,4 +61,4 @@ class BetterNavigator extends DataExtension {
 		}
 		return false;
 	}
-}
\ No newline at end of file
+}