diff --git a/src/Block/BannerBlock.php b/src/Block/BannerBlock.php
index d8b8d65..4f4ef39 100644
--- a/src/Block/BannerBlock.php
+++ b/src/Block/BannerBlock.php
@@ -5,7 +5,7 @@
 use SilverStripe\CMS\Model\SiteTree;
 use SilverStripe\ElementalFileBlock\Block\FileBlock;
 use SilverStripe\Forms\FieldList;
-use SilverStripe\View\ArrayData;
+use SilverStripe\Model\ArrayData;
 
 class BannerBlock extends FileBlock
 {
diff --git a/src/Form/BlockLinkField.php b/src/Form/BlockLinkField.php
index 3dbb98a..580a300 100644
--- a/src/Form/BlockLinkField.php
+++ b/src/Form/BlockLinkField.php
@@ -6,7 +6,7 @@
 use SilverStripe\Core\Convert;
 use SilverStripe\Forms\FormField;
 use SilverStripe\ORM\DataObject;
-use SilverStripe\View\ArrayData;
+use SilverStripe\Model\ArrayData;
 
 class BlockLinkField extends FormField
 {
diff --git a/tests/Block/BannerBlockTest.php b/tests/Block/BannerBlockTest.php
index 9785619..8381257 100644
--- a/tests/Block/BannerBlockTest.php
+++ b/tests/Block/BannerBlockTest.php
@@ -6,7 +6,7 @@
 use SilverStripe\Dev\SapphireTest;
 use SilverStripe\ElementalBannerBlock\Block\BannerBlock;
 use SilverStripe\Forms\HTMLEditor\TinyMCEConfig;
-use SilverStripe\View\ArrayData;
+use SilverStripe\Model\ArrayData;
 use SilverStripe\View\Requirements;
 
 class BannerBlockTest extends SapphireTest
diff --git a/tests/Form/BlockLinkFieldTest.php b/tests/Form/BlockLinkFieldTest.php
index d9ae2dc..de60da8 100644
--- a/tests/Form/BlockLinkFieldTest.php
+++ b/tests/Form/BlockLinkFieldTest.php
@@ -4,7 +4,7 @@
 
 use SilverStripe\Dev\SapphireTest;
 use SilverStripe\ElementalBannerBlock\Form\BlockLinkField;
-use SilverStripe\View\ArrayData;
+use SilverStripe\Model\ArrayData;
 
 class BlockLinkFieldTest extends SapphireTest
 {