diff --git a/tests/MultiFormObjectDecoratorTest.php b/tests/MultiFormObjectDecoratorTest.php
index f825b2a..9683f95 100644
--- a/tests/MultiFormObjectDecoratorTest.php
+++ b/tests/MultiFormObjectDecoratorTest.php
@@ -3,17 +3,18 @@
 namespace SilverStripe\MultiForm\Tests;
 
 use SilverStripe\Dev\SapphireTest;
+use SilverStripe\MultiForm\Extensions\MultiFormObjectDecorator;
 
 class MultiFormObjectDecoratorTest extends SapphireTest
 {
     protected static $fixture_file = 'MultiFormObjectDecoratorTest.yml';
 
-    protected $requiredExtensions = [
-        'MultiFormObjectDecoratorDataObject' => ['MultiFormObjectDecorator']
+    protected static $required_extensions = [
+        MultiFormObjectDecoratorDataObject::class => [MultiFormObjectDecorator::class]
     ];
 
     protected $extraDataObjects = [
-        'MultiFormObjectDecoratorDataObject'
+        MultiFormObjectDecoratorDataObject::class
     ];
 
     public function testTemporaryDataFilteredQuery()
diff --git a/tests/Stubs/MultiFormObjectDecoratorDataObject.php b/tests/Stubs/MultiFormObjectDecoratorDataObject.php
index f2d3d0f..7b05c4d 100644
--- a/tests/Stubs/MultiFormObjectDecoratorDataObject.php
+++ b/tests/Stubs/MultiFormObjectDecoratorDataObject.php
@@ -10,4 +10,6 @@ class MultiFormObjectDecoratorDataObject extends DataObject implements TestOnly
     private static $db = [
         'Name' => 'Varchar'
     ];
+
+    private static $table_name = 'MultiFormObjectDecoratorDataObject';
 }
diff --git a/tests/Stubs/MultiFormTestController.php b/tests/Stubs/MultiFormTestController.php
index b7b6a2f..ee0a86b 100644
--- a/tests/Stubs/MultiFormTestController.php
+++ b/tests/Stubs/MultiFormTestController.php
@@ -11,7 +11,7 @@ use SilverStripe\Dev\TestOnly;
  */
 class MultiFormTestController extends Controller implements TestOnly
 {
-    public function Link()
+    public function Link($action = null)
     {
         return self::class;
     }