diff --git a/composer.json b/composer.json
index 761bcce..4de41ce 100644
--- a/composer.json
+++ b/composer.json
@@ -1,6 +1,6 @@
 {
 	"name": "mage2pro/yandex-kassa"
-	,"version": "0.2.0"
+	,"version": "0.2.1"
 	,"description": "The «Yandex.Kassa» (Яндекс.Касса, Yandex.Checkout) payment extension for Magento 2."
 	,"type": "magento2-module"
 	,"homepage": "https://mage2.pro/c/extensions/yandex-kassa"
diff --git a/etc/di.xml b/etc/di.xml
new file mode 100644
index 0000000..c1412bc
--- /dev/null
+++ b/etc/di.xml
@@ -0,0 +1,8 @@
+<?xml version='1.0'?>
+<config
+	xmlns:xsi='http://www.w3.org/2001/XMLSchema-instance'
+	xsi:noNamespaceSchemaLocation='urn:magento:framework:ObjectManager/etc/config.xsd'
+>
+	<virtualType name='Dfe\YandexKassa\Controller\CustomerReturn\Index' type='Df\Payment\CustomerReturn'/>
+	<virtualType name='Dfe\YandexKassa\Controller\Index\Index' type='Df\Payment\W\Action'/>
+</config>
\ No newline at end of file