diff --git a/Controller/AbstractDataTablesController.php b/Controller/AbstractDataTablesController.php
index 0641f4e7..0ed52bf3 100644
--- a/Controller/AbstractDataTablesController.php
+++ b/Controller/AbstractDataTablesController.php
@@ -315,6 +315,7 @@ protected function getDataTablesWrapper(DataTablesProviderInterface $dtProvider)
         // Initialize the wrapper.
         $dtWrapper = new DataTablesWrapper($dtProvider->getMethod(), $url, $dtProvider->getName());
         $dtWrapper->getMapping()->setPrefix($dtProvider->getPrefix());
+        $dtWrapper->setProvider($dtProvider);
 
         // Handle each column.
         foreach ($dtProvider->getColumns() as $dtColumn) {