diff --git a/tests/CollectionTest.php b/tests/CollectionTest.php
index f011e829..23821615 100644
--- a/tests/CollectionTest.php
+++ b/tests/CollectionTest.php
@@ -207,9 +207,9 @@ public function testMultipleCollections()
 
             $inner = (new Collection('innerCollection'))
                 ->setLabel('Inner Collection')
-                ->setAddElement(new SubmitButtonElement('inner_add_trigger', [
+                ->setAddElement('submitButton', 'inner_add_trigger', [
                     'label' => 'Inner Add Trigger'
-                ]));
+                ]);
 
             $inner->onAssembleGroup(function ($innerGroup, $innerAddElement, $innerRemoveElement) {
                 $innerGroup->addElement($innerAddElement);