diff --git a/src/Model/Comment.php b/src/Model/Comment.php
index 31c2443..cdb796c 100755
--- a/src/Model/Comment.php
+++ b/src/Model/Comment.php
@@ -26,6 +26,7 @@ use SilverStripe\ORM\DB;
 use SilverStripe\ORM\PaginatedList;
 use SilverStripe\Security\Member;
 use SilverStripe\Security\Permission;
+use SilverStripe\CMS\Model\SiteTree;
 
 /**
  * Represents a single comment object.