diff --git a/src/TestSessionEnvironment.php b/src/TestSessionEnvironment.php
index c5fdedd..7cdb5df 100644
--- a/src/TestSessionEnvironment.php
+++ b/src/TestSessionEnvironment.php
@@ -21,7 +21,7 @@
 use SilverStripe\ORM\Connect\TempDatabase;
 use SilverStripe\ORM\DB;
 use SilverStripe\ORM\FieldType\DBDatetime;
-use SilverStripe\Versioned\Versioned;
+use SilverStripe\Versioned\Mode\Versioned;
 use Symfony\Component\Mailer\MailerInterface;
 use stdClass;