diff --git a/src/DependencyInjection/Configuration.php b/src/DependencyInjection/Configuration.php
index 8098829..1882794 100644
--- a/src/DependencyInjection/Configuration.php
+++ b/src/DependencyInjection/Configuration.php
@@ -31,8 +31,12 @@ class Configuration implements ConfigurationInterface
      */
     public function getConfigTreeBuilder()
     {
-        $treeBuilder = new TreeBuilder();
-        $treeBuilder->root('sonata_datagrid', 'array');
+        $treeBuilder = new TreeBuilder('sonata_datagrid');
+
+        // Keep compatibility with symfony/config < 4.2
+        if (!\method_exists($treeBuilder, 'getRootNode')) {
+            $treeBuilder->root('sonata_datagrid');
+        }
 
         return $treeBuilder;
     }