diff --git a/src/Resources/config/doctrine/Ad.orm.xml b/src/Resources/config/doctrine/Ad.orm.xml
index 778e45a..da710f0 100644
--- a/src/Resources/config/doctrine/Ad.orm.xml
+++ b/src/Resources/config/doctrine/Ad.orm.xml
@@ -16,7 +16,7 @@
         <field name="priority" column="priority" type="integer"/>
         <field name="code" column="code"/>
 
-        <many-to-many field="banners" mapped-by="ads" target-entity="BitBag\SyliusBannerPlugin\Entity\Banner"/>
+        <many-to-many field="banners" mapped-by="ads" target-entity="BitBag\SyliusBannerPlugin\Entity\BannerInterface"/>
 
     </mapped-superclass>
 </doctrine-mapping>
diff --git a/src/Resources/config/doctrine/Banner.orm.xml b/src/Resources/config/doctrine/Banner.orm.xml
index 9c9e83b..88ef792 100644
--- a/src/Resources/config/doctrine/Banner.orm.xml
+++ b/src/Resources/config/doctrine/Banner.orm.xml
@@ -24,11 +24,11 @@
             <join-column name="locale_id" referenced-column-name="id"/>
         </many-to-one>
 
-        <many-to-one field="section" target-entity="BitBag\SyliusBannerPlugin\Entity\Section" inversed-by="banners">
+        <many-to-one field="section" target-entity="BitBag\SyliusBannerPlugin\Entity\SectionInterface" inversed-by="banners">
             <join-column name="section_id" referenced-column-name="id"/>
         </many-to-one>
 
-        <many-to-many field="ads" inversed-by="banners" target-entity="BitBag\SyliusBannerPlugin\Entity\Ad">
+        <many-to-many field="ads" inversed-by="banners" target-entity="BitBag\SyliusBannerPlugin\Entity\AdInterface">
             <join-table name="bitbag_banners_ads">
                 <join-columns>
                     <join-column name="banner_id"/>
diff --git a/src/Resources/config/doctrine/Section.orm.xml b/src/Resources/config/doctrine/Section.orm.xml
index e86dd7f..24a6cee 100644
--- a/src/Resources/config/doctrine/Section.orm.xml
+++ b/src/Resources/config/doctrine/Section.orm.xml
@@ -14,7 +14,7 @@
         <field name="width" column="width" type="integer" nullable="true"/>
         <field name="height" column="height" type="integer" nullable="true"/>
 
-        <one-to-many field="banners" target-entity="BitBag\SyliusBannerPlugin\Entity\Banner" mapped-by="section"/>
+        <one-to-many field="banners" target-entity="BitBag\SyliusBannerPlugin\Entity\BannerInterface" mapped-by="section"/>
     </mapped-superclass>
 </doctrine-mapping>