diff --git a/etc/di.xml b/etc/di.xml
index 89aabe6..a3beae9 100644
--- a/etc/di.xml
+++ b/etc/di.xml
@@ -131,7 +131,7 @@
         </arguments>
     </virtualType>
 
-    <virtualType name="Macopedia\Allegro\Model\ResourceModel\Reservation\Collection" type="Magento\Framework\View\Element\UiComponent\DataProvider\SearchResult">
+    <virtualType name="Macopedia\Allegro\Model\ResourceModel\Reservation\Collection\Grid" type="Magento\Framework\View\Element\UiComponent\DataProvider\SearchResult">
         <arguments>
             <argument name="mainTable" xsi:type="string">allegro_reservations</argument>
             <argument name="resourceModel" xsi:type="string">Macopedia\Allegro\Model\ResourceModel\Reservation</argument>