diff --git a/etc/di.xml b/etc/di.xml
index afd2fef27..1589b9d59 100644
--- a/etc/di.xml
+++ b/etc/di.xml
@@ -244,6 +244,7 @@
                 <item name="/tests/test/before/getData/requiredEntity" xsi:type="string">createDataKey</item>
                 <item name="/tests/test/after/getData/requiredEntity" xsi:type="string">createDataKey</item>
                 <item name="/tests/test/actionGroup/argument" xsi:type="string">name</item>
+                <item name="/tests/test/(before|after)/actionGroup/argument" xsi:type="string">name</item>
             </argument>
             <argument name="numericArrays" xsi:type="array">
                 <item name="/tests/test/annotations/features" xsi:type="string">/tests/test/annotations/features</item>