diff --git a/src/Controllers/ShareDraftController.php b/src/Controllers/ShareDraftController.php
index 2753417..91fdda8 100644
--- a/src/Controllers/ShareDraftController.php
+++ b/src/Controllers/ShareDraftController.php
@@ -18,7 +18,7 @@
 use SilverStripe\ShareDraftContent\Extensions\ShareDraftContentSiteTreeExtension;
 use SilverStripe\ShareDraftContent\Models\ShareToken;
 use SilverStripe\Versioned\Versioned;
-use SilverStripe\View\ArrayData;
+use SilverStripe\Model\ArrayData;
 use SilverStripe\View\Requirements;
 
 class ShareDraftController extends Controller