diff --git a/DependencyInjection/Configuration.php b/DependencyInjection/Configuration.php
index 017342b..a2e4791 100644
--- a/DependencyInjection/Configuration.php
+++ b/DependencyInjection/Configuration.php
@@ -20,8 +20,14 @@ class Configuration implements ConfigurationInterface
      */
     public function getConfigTreeBuilder()
     {
-        $treeBuilder = new TreeBuilder();
-        $rootNode = $treeBuilder->root('doctrine_migrations', 'array');
+        $treeBuilder = new TreeBuilder('doctrine_migrations');
+
+        if (method_exists($treeBuilder, 'getRootNode')) {
+            $rootNode = $treeBuilder->getRootNode();
+        } else {
+            // BC layer for symfony/config 4.1 and older
+            $rootNode = $treeBuilder->root('doctrine_migrations', 'array');
+        }
 
         $organizeMigrationModes = $this->getOrganizeMigrationsModes();