diff --git a/src/Sylius/Bundle/UiBundle/Tests/Functional/Kernel.php b/src/Sylius/Bundle/UiBundle/Tests/Functional/Kernel.php
index 0f4a947b6e..648301fb42 100644
--- a/src/Sylius/Bundle/UiBundle/Tests/Functional/Kernel.php
+++ b/src/Sylius/Bundle/UiBundle/Tests/Functional/Kernel.php
@@ -132,9 +132,11 @@ final class Kernel extends HttpKernel
             'template_event_with_context' => [
                 'blocks' => [
                     'block' => [
-                        'component' => 'SomeTwigComponent',
-                        'context' => [
-                            'baz' => 'qux',
+                        'component' => [
+                            'name' => 'SomeTwigComponent',
+                            'inputs' => [
+                                'context' => 'expr:context',
+                            ],
                         ],
                     ],
                 ],
diff --git a/src/Sylius/Bundle/UiBundle/Tests/Functional/TemplateEventTest.php b/src/Sylius/Bundle/UiBundle/Tests/Functional/TemplateEventTest.php
index fc0ddf6383..6acb7949b0 100644
--- a/src/Sylius/Bundle/UiBundle/Tests/Functional/TemplateEventTest.php
+++ b/src/Sylius/Bundle/UiBundle/Tests/Functional/TemplateEventTest.php
@@ -114,7 +114,7 @@ final class TemplateEventTest extends KernelTestCase
             '<!-- BEGIN EVENT | event name: "template_event_with_context" -->',
             '<!-- BEGIN BLOCK | event name: "template_event_with_context", block name: "block", component: "SomeTwigComponent", priority: 0 -->',
             'Hello from the Twig Component :)',
-            'Context: baz=qux, foo=bar',
+            'Context: foo=bar',
             '<!-- END BLOCK | event name: "template_event_with_context", block name: "block" -->',
             '<!-- END EVENT | event name: "template_event_with_context" -->',
         ];
diff --git a/src/Sylius/Bundle/UiBundle/spec/Renderer/TwigComponentBlockRendererSpec.php b/src/Sylius/Bundle/UiBundle/spec/Renderer/TwigComponentBlockRendererSpec.php
index d29974d217..602f3eb1b8 100644
--- a/src/Sylius/Bundle/UiBundle/spec/Renderer/TwigComponentBlockRendererSpec.php
+++ b/src/Sylius/Bundle/UiBundle/spec/Renderer/TwigComponentBlockRendererSpec.php
@@ -64,7 +64,7 @@ final class TwigComponentBlockRendererSpec extends ObjectBehavior
             'Component',
             [
                 'foo' => 'bar',
-                'bar' => 'expr:foo',
+                'bar' => 'expr:baz',
                 'nested' => [
                     'foo' => 'expr:bar',
                     'bar' => 'expr:baz',
@@ -90,7 +90,7 @@ final class TwigComponentBlockRendererSpec extends ObjectBehavior
         $componentRenderer
             ->createAndRender('Component', [
                 'foo' => 'bar',
-                'bar' => 'bar',
+                'bar' => 'qux',
                 'nested' => [
                     'foo' => 'baz',
                     'bar' => 'qux',