diff --git a/code/Company.php b/code/Company.php
index 5e27c21..278a13e 100644
--- a/code/Company.php
+++ b/code/Company.php
@@ -9,8 +9,8 @@
 use SilverStripe\ORM\DataObject;
 use SilverStripe\ORM\DB;
 use SilverStripe\AssetAdmin\Forms\UploadField;
-use SilverStripe\Versioned\RecursivePublishable;
-use SilverStripe\Versioned\Versioned;
+use SilverStripe\Versioned\Staged\RecursivePublishable;
+use SilverStripe\Versioned\Mode\Versioned;
 use RelationFieldsTestPage;
 use GridFieldTestPage;
 use SilverStripe\Forms\RequiredFields;