diff --git a/composer.json b/composer.json
index 9615714..3bdf2c9 100644
--- a/composer.json
+++ b/composer.json
@@ -1,6 +1,6 @@
 {
 	"name": "mage2pro/payfort"
-	,"version": "0.0.3"
+	,"version": "0.0.4"
 	,"description": "PayFort integration with Magento 2"
 	,"type": "magento2-module"
 	,"homepage": "https://mage2.pro/c/extensions/payfort"
diff --git a/etc/di.xml b/etc/di.xml
deleted file mode 100644
index 774c785..0000000
--- a/etc/di.xml
+++ /dev/null
@@ -1,7 +0,0 @@
-<?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\PayFort\Controller\Index\Index' type='Df\Payment\W\Action'/>
-</config>
\ No newline at end of file
diff --git a/etc/frontend/di.xml b/etc/frontend/di.xml
index 83181c4..0171a00 100644
--- a/etc/frontend/di.xml
+++ b/etc/frontend/di.xml
@@ -20,4 +20,5 @@
 	<virtualType name='Dfe\PayFort\ConfigProvider' type='Df\StripeClone\ConfigProvider'>
 		<arguments><argument name='module' xsi:type='string'>Dfe_PayFort</argument></arguments>
 	</virtualType>
+	<virtualType name='Dfe\PayFort\Controller\Index\Index' type='Df\Payment\W\Action'/>
 </config>
\ No newline at end of file