diff --git a/src/DependencyInjection/Configuration.php b/src/DependencyInjection/Configuration.php
index ae7a7ce..7362a2b 100644
--- a/src/DependencyInjection/Configuration.php
+++ b/src/DependencyInjection/Configuration.php
@@ -21,8 +21,8 @@ class Configuration implements ConfigurationInterface
      */
     public function getConfigTreeBuilder()
     {
-        $treeBuilder = new TreeBuilder();
-        $rootNode = $treeBuilder->root('swagger');
+        $treeBuilder = new TreeBuilder('swagger');
+        $rootNode =  \method_exists(TreeBuilder::class, 'getRootNode') ? $treeBuilder->getRootNode() : $treeBuilder->root('swagger');
 
         $rootNode
             ->children()