diff --git a/tests/ErrorPageFileExtensionTest.php b/tests/ErrorPageFileExtensionTest.php
index a41341b..43bfa6f 100644
--- a/tests/ErrorPageFileExtensionTest.php
+++ b/tests/ErrorPageFileExtensionTest.php
@@ -10,7 +10,7 @@
 use SilverStripe\ErrorPage\ErrorPage;
 use SilverStripe\Security\InheritedPermissions;
 use SilverStripe\Security\Security;
-use SilverStripe\Versioned\Versioned;
+use SilverStripe\Versioned\Mode\Versioned;
 use SilverStripe\View\Parsers\ShortcodeParser;
 
 class ErrorPageFileExtensionTest extends SapphireTest
diff --git a/tests/ErrorPageTest.php b/tests/ErrorPageTest.php
index 4974313..5dedb70 100644
--- a/tests/ErrorPageTest.php
+++ b/tests/ErrorPageTest.php
@@ -16,7 +16,7 @@
 use SilverStripe\ErrorPage\ErrorPage;
 use SilverStripe\ORM\DB;
 use SilverStripe\Core\Validation\ValidationException;
-use SilverStripe\Versioned\Versioned;
+use SilverStripe\Versioned\Mode\Versioned;
 use SilverStripe\View\SSViewer;
 use PHPUnit\Framework\Attributes\DataProvider;